如何自定义cmap
2023-11-10 本文已影响0人
可能性之兽
在Python的Matplotlib库中，可以使用LinearSegmentedColormap函数来创建自定义的颜色映射（colormap）
import matplotlib.pyplot as plt
import numpy as np
from matplotlib.colors import LinearSegmentedColormap
# 创建一个随机数据数组
data = np.random.rand(10, 10)
# 定义颜色从红色到绿色
colors = ["red", "green"]
# 创建自定义颜色映射
cmap = LinearSegmentedColormap.from_list("my_colormap", colors)
# 使用自定义颜色映射绘制图像
plt.pcolormesh(data, cmap=cmap)
plt.colorbar() # 显示颜色条
plt.show()
