matplotlib.colorbar应显示单色红色光谱

Question

首先，我在[0,1]上规范化我的数据，将相应的值分配给红色光谱的颜色。

reds = (data/data.max()).flatten()
rgb_data = np.array([(r,0,0)for r in reds]).reshape((x.max(),y.max(),3))

我创建了一个色彩图

my_cmap = mcolors.ListedColormap(rgb_data)
cax = ax.imshow(rgb_data)

问题在于颜色栏

cbar = colorbar(cax)

也跨越颜色，绿色和蓝色不等于0.

如何定制它，从黑色到深红色，而不会超越绿色和蓝色？

Answer 1

你几乎没错了

cax = ax.imshow(rgb_data, cmap = my_cmap)

您定义了自己的色彩映射，但您还必须将其作为参数传递。