matplotlib颜色条用于分散

Question

我正在处理的数据具有3个绘图参数：x，y，c。如何为散点图创建自定义颜色值？

扩展这个例子，我想做的是：

import matplotlib

import matplotlib.pyplot as plt

cm = matplotlib.cm.get_cmap('RdYlBu')

colors=[cm(1.*i/20) for i in range(20)]

xy = range(20)

plt.subplot(111)

colorlist=[colors[x/2] for x in xy] #actually some other non-linear relationship

plt.scatter(xy, xy, c=colorlist, s=35, vmin=0, vmax=20)

plt.colorbar()

plt.show()

但结果是 TypeError: You must first set_array for mappable

Answer 1

这是添加颜色条的OOP方法：fig, ax = plt.subplots()im = ax.scatter(x, y, c=c)fig.colorbar(im, ax=ax)