Python 3 使用 Matplotlib 添加颜色条

Question

我已经对向绘图添加颜色条进行了大量研究，但我仍然对如何添加颜色条感到困惑。我见过的例子使用了不同的方法，这让我感到困惑，因为我不明白什么是“正确”的方法。

我已经看到有一个 colorbar 方法和一个 colorbar() 函数，那么应该使用什么来简单地添加一个颜色条？

一些例子是这样做的:

fig,ax = plt.subplots()
s = ax.scatter(x,y,cmap = coolwarm)
matplotlib.colorbar.ColorbarBase(ax=ax, cmap=coolwarm, values=sorted(v),
                             orientation="horizontal")

而其他一些人只是简单地调用该函数:

fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)
image = np.random.poisson(10., (100, 80))
i = ax.imshow(image, interpolation='nearest')
fig.colorbar(i)

我可能在这里遗漏了一些东西，但我只是没有看到它们是如何创建颜色条的(我只是复制了颜色条的代码并排除了数据的代码)。

我的问题很简单:向绘图添加颜色条的最简单方法是什么？
谢谢!

Answer 1

您引用的第一个示例创建了 ColorbarBase 的一个实例。这通常不是推荐的方式；可能有一些异常(exception)，但总的来说，绝对没有理由使用它。

您引用的第二个示例是在图形内创建颜色条的一种甚至方法。使用它，您就处于保存状态。使用 colorbar图形实例的方法明确在哪个图形中放置颜色条并提供相应的 ScalarMappable(在本例中为 AxesImage)确保颜色条使用来自该 ScalarMappable 的正确颜色。

fig, ax = plt.subplots()
im = ax.imshow(image)
fig.colorbar(im)

或者

fig, ax = plt.subplots()
sc = ax.scatter(x,y, c=something)
fig.colorbar(sc)

有一个更简单的方法，那就是简单地调用

plt.colorbar()

但是请注意，这可能会导致混淆，因为它会尝试自动确定应为其创建颜色条的绘图。因此，它有可能失败，我不建议使用它。