将颜色条作为图例添加到 matplotlib 散点图（多个子图，多个散点图）

Question

我有几个要添加单个颜色条的子图。每个子图由 7 个散点组成。我找到了关于如何添加颜色条的建议，但它们主要与每个散点的值有关，而不是与行本身有关。

代表性示例代码：

import numpy as np
from matplotlib import pyplot as plt

x = range(50)
scales = np.linspace(0, 2, 7)
locs = range(4)
cmap = plt.get_cmap("Spectral")
for s_plot in range(4):
    plt.subplot(2, 2, s_plot+1)
    color = iter(cmap(np.linspace(0, 1, len(scales))))
    for scale in scales:
        c = next(color)
        y = np.random.normal(loc=locs[s_plot], scale=scale, size=50)
        plt.scatter(x, y, c=c, s=5)
        plt.title("Mean = {:d}".format(locs[s_plot]))
plt.subplots_adjust(hspace=0.4)
plt.show()

上面的例子给出：

我想要的颜色条看起来像这样（假的，放在情节旁边）：

所以颜色条没有描述我的散点的值，而是迭代的不同“行”（在这种情况下：不同的比例）。在有助于将点与比例匹配的示例中。

我尝试的是一个简单的

plt.colorbar()

在完成每个子图后调用一次。但我也得到了 TypeError: You must first set_array for mappable ，因为它是我想为其创建颜色图的不同比例，我也尝试过

plt.colorbar(scales) 

返回：AttributeError: 'numpy.ndarray' object has no attribute 'autoscale_None'.

我目前缺乏关于如何继续进行的方向。编辑：我被标记为 scatter的matplotlib colorbar 的可能副本。我已经发现了这个问题，但它对我的问题没有帮助。在我的情况下，我需要一个独立于 z 值的颜色图，但只会指示“行号”或“分散行”，或者您想如何称呼它（相当于 a 中的“行” plt.plot）。

Answer 1

颜色条需要 aScalarMappable作为输入。因此，如果您在情节中创建的所有内容都不适合该内容，则您可以自己创建它。

import numpy as np
from matplotlib import pyplot as plt
from matplotlib.cm import ScalarMappable

x = range(50)
scales = np.linspace(0, 2, 7)
locs = range(4)
cmap = plt.get_cmap("Spectral")
norm = plt.Normalize(scales.min(), scales.max())

fig, axes = plt.subplots(2,2, constrained_layout=True, sharey=True)

for s_plot, ax in enumerate(axes.flat):
    for scale in scales:
        y = np.random.normal(loc=locs[s_plot], scale=scale, size=50)
        sc = ax.scatter(x, y, c=[cmap(norm(scale))], s=5)
        ax.set_title("Mean = {:d}".format(locs[s_plot]))

sm =  ScalarMappable(norm=norm, cmap=cmap)
sm.set_array([])
cbar = fig.colorbar(sm, ax=axes[:,1])
cbar.ax.set_title("scale")

plt.show()