如何给matplotlib图片加上渐变背景

添加渐变背景#

有时候，我们希望在绘图时添加渐变背景，这样可以使图像更加美观。我们可以使用fig.figimage方法来添加渐变背景。fig.figimage方法可以在figure上添加一个图像，参数cmap表示颜色映射，参数resize表示是否调整大小，参数origin表示图像的原点位置。我们可以使用LinearSegmentedColormap.from_list()方法来创建一个颜色映射。

注意：如果使用fig.figimage方法添加渐变背景，必须保证绘图dpi和保存dpi一致，否则会出现背景不一致的情况。

1
from matplotlib.colors import LinearSegmentedColormap
2
fig = plt.figure(figsize=(cm(15), cm(10)),dpi=500)
3
ax = plt.gca()
4

5
# 创建自定义渐变色彩映射
6
# 斜向上，白色到蓝色渐变
7
colors = ["white", "#ffe2e2"]  # 白色到蓝色
8
cmap = LinearSegmentedColormap.from_list("custom_gradient", colors, N=255)
9

10
# 为整个figure设置渐变背景
11
# 首先根据dpi算出gradient宽度和高度
12
gradient_width=int(fig.dpi*fig.get_figwidth())
13
gradient_height=int(fig.dpi*fig.get_figheight())
14
# 然后生成斜向上的渐变（左下到右上，白到蓝）
15
gradient = np.linspace(0, 1, gradient_width)
16
gradient = np.tile(gradient, (gradient_height, 1))
17
gradient = (gradient + np.linspace(0, 1, gradient_height)[:, None]) / 2
18

19
# 将渐变设置为figure的背景
20
fig.figimage(cmap(gradient), resize=False,origin="lower")
21
# 确保坐标轴显示在渐变之上
22
ax.set_zorder(10)
23
ax.patch.set_alpha(0)  # 使坐标轴背景透明，以便看到渐变

添加渐变背景#

之后就可以加入正常的绘图代码了，加入绘图代码后的绘图结果如下所示#

目录