您显示的颜色图看起来只是黑色、蓝色、红色和黄色之间的插值。这将通过
创建
matplotlib.colors.LinearSegmentedColormap.from_list("", ["black", "blue", "red", "yellow"])
但是,我不建议使用这样的颜色图。请改用"viridis"、"magma"、"plasma" 或"inferno" 中的任何一个。这些在感知上是统一的,因此不会产生不需要的伪影。
下面是插值地图和"magma" 之间的比较。如您所见,插值地图创建了“环”,因此查看者肯定会误解所绘制的形状。
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import colors as mcolors
x = np.linspace(-2,2, 101)
X,Y = np.meshgrid(x,x)
Z = np.exp(-X**2-Y**2)
fig, (ax1, ax2) = plt.subplots(ncols=2)
im1 = ax1.imshow(Z, cmap="magma")
cmap = mcolors.LinearSegmentedColormap.from_list("", ["black", "blue", "red", "yellow"])
im2 = ax2.imshow(Z, cmap=cmap)
fig.colorbar(im1, ax=ax1, orientation="horizontal")
fig.colorbar(im2, ax=ax2, orientation="horizontal")
plt.show()
进一步阅读