python - 多个散点图的一个颜色条

Question

我正在使用 matplotlib 库在一个图上绘制同一变量的多个图形。 我不是在寻找子图的colorbar，这是主要的搜索 Material 。我绘制了多个 scatter，但 colorbar 仅设置为我绘制的最后一个散点的值。

部分代码如下:

plt.scatter(x1, y1, c=z1,cmap='viridis_r',marker='s')
plt.scatter(x2, y2, c=z2,cmap='viridis_r',marker='o')
plt.scatter(x3, y3, c=z3,cmap='viridis_r',marker='^')
plt.colorbar().set_label('Wind speed',rotation=270)

Answer 1

它需要一些额外的工作:

• 您必须获得 c 的最小值和最大值(颜色条值)
• 您必须攀登每个散点图

首先是最小值和最大值:

zs = np.concatenate([z1, z2, z3], axis=0)
min_, max_ = zs.min(), zs.max()

然后用clim绘制散点图:

plt.scatter(x1, y1, c=z1,cmap='viridis_r',marker='s')
plt.clim(min_, max_)
plt.scatter(x2, y2, c=z2,cmap='viridis_r',marker='o')
plt.clim(min_, max_)
plt.scatter(x3, y3, c=z3,cmap='viridis_r',marker='^')
plt.clim(min_, max_)
plt.colorbar().set_label('Wind speed',rotation=270)

对于一个非常简单的数据集:

x1, x2, x3 = [1,2,3], [2,3,4], [3,4,5]
y1 = y2 = y3 = [1, 2, 3]
z1, z2, z3 = [1,2,3], [4,5,6], [7,8,9]