我正在处理具有 3 个绘图参数的数据:x、y、c。如何为散点图创建自定义颜色值?
扩展此 example我正在尝试:
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
最佳答案
来自matplotlib docs on scatter 1 :
cmap is only used if c is an array of floats
所以 colorlist 需要是一个浮点列表,而不是你现在拥有的元组列表。 plt.colorbar() 想要一个可映射的对象,例如 plt.scatter() 返回的 CircleCollection。 然后 vmin 和 vmax 可以控制颜色条的限制。 vmin/vmax 之外的东西获取端点的颜色。
这对你有什么作用?
import matplotlib.pyplot as plt
cm = plt.cm.get_cmap('RdYlBu')
xy = range(20)
z = xy
sc = plt.scatter(xy, xy, c=z, vmin=0, vmax=20, s=35, cmap=cm)
plt.colorbar(sc)
plt.show()
关于python - 用于散点图的 matplotlib 颜色条,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/6063876/