散点图颜色条 - Matplotlib

Question

我正在尝试显示我的散点图的颜色条，但我不断收到错误消息：

TypeError: You must first set_array for mappable

这就是我正在做的情节：

# Just plotting the values of data that are nonzero 
x_data = numpy.nonzero(data)[0] # x coordinates
y_data = numpy.nonzero(data)[1] # y coordinates

# Mapping the values to RGBA colors
data = plt.cm.jet(data[x_data, y_data])

pts = plt.scatter(x_data, y_data, marker='s', color=data)

plt.colorbar(pts)

如果我评论 plt.colorbar(pts) 行，我得到了正确的绘图，但我也想绘制颜色条。

提前谢谢你。

Answer 1

您正在传递特定的 rgb 值，因此 matplotlib 无法构建颜色图，因为它不知道它与您的原始数据的关系。

不要将值映射到 RGB 颜色，而是让 scatter 为您处理。

代替：

# Mapping the values to RGBA colors
data = plt.cm.jet(data[x_data, y_data])

pts = plt.scatter(x_data, y_data, marker='s', color=data)

做：

pts = plt.scatter(x_data, y_data, marker='s', c=data[x_data, y_data])

（只需将您最初传递给plt.cm.jet的内容传递给c。）

然后您就可以正常构建颜色图了。具体的错误是告诉您颜色是手动设置的，而不是通过set_array（处理将数据值数组映射到RGB）设置的。