制作简单的颜色条

Question

我有一些数据，想用散点图来展示，也就是(x,y)这样的点。不过，每个点我还有一个第三个值，比如说那个点的温度。我想在y轴旁边加一个颜色条，用来表示这些点的温度变化。我查过matplotlib网站上的文档，但还是搞不明白。

x = [1,2,3,4]
y = [10,20,30,40]
and Temp = [100,200,300,400]

我用plt.plot(x,y)来画图，但对于颜色条，plt.colorbar里面应该放什么呢？

这个问题其实很简单，我相信，但我需要尽快找到解决办法，现在有点困难，所以任何建议都非常感谢。谢谢！

Answer 1

你可以直接使用 plt.colorbar()，不需要传入任何参数，因为 pyplot 知道你在做的散点图。正如之前有评论的人提到的，这个例子用 plt.scatter 会更简单：

import matplotlib.pyplot as plt
x = [1,2,3,4]
y = [10,20,30,40]
Temp = [100,200,300,400]

plt.scatter(x,y,c=Temp)
plt.colorbar()
plt.show()

在 plt.scatter 里，你可以放入各种奇奇怪怪的参数。可以看看文档，还有 matplotlib 的缩略图来获取灵感。祝你在使用 matplotlib 的过程中愉快！