创建matplotlib渐变图例

要创建一个matplotlib渐变图例，可以使用colorbar函数结合ListedColormap和LinearSegmentedColormap来定义颜色映射。

以下是一个创建渐变图例的示例代码：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap, LinearSegmentedColormap

# 创建一个线性分段的颜色映射
colors = ['#0000ff', '#00ffff', '#ffffff', '#ffff00', '#ff0000']
cmap = LinearSegmentedColormap.from_list('my_cmap', colors, N=256)

# 生成一些随机数据
data = np.random.rand(10, 10)

# 绘制热力图
plt.imshow(data, cmap=cmap)

# 创建渐变图例
bounds = np.linspace(0, 1, len(colors))
norm = plt.Normalize(bounds.min(), bounds.max())
cb = plt.colorbar(plt.cm.ScalarMappable(norm=norm, cmap=cmap), ticks=bounds)
cb.set_ticklabels(['Low', 'Medium', 'High'])  # 设置标签

plt.show()

在这个示例中，首先创建一个由自定义颜色组成的线性分段的颜色映射。然后使用imshow函数绘制热力图，并将颜色映射设置为之前创建的线性分段的颜色映射。最后，使用colorbar函数创建一个渐变图例，其中norm参数用于将颜色映射的范围映射到0-1之间，ticks参数用于设置刻度位置，set_ticklabels方法用于设置刻度标签。

可以根据需要修改colors列表和set_ticklabels方法中的标签来自定义渐变图例的颜色和标签。