matplotlib颜色条用于分散

Question

我正在处理的数据具有3个绘图参数：x，y，c。如何为散点图创建自定义颜色值？扩展这个例子，我想做的是：import matplotlibimport matplotlib.pyplot as pltcm = 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 relationshipplt.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)