Matplotlib：在一组 arrays 上循环散布 plot，具有一致的散点大小和颜色条

Question

我正在尝试使用 dataframe 系列 x 和 y 生成散点图 plot，并使用 dataframe 系列 z 生成散点图数据点的大小。

我应该提一下，我遍历了一组 x、y 和 z arrays，并在循环外添加了颜色 plot。

我看到每次迭代都会生成散点大小和颜色条，因此散点大小与 plot 中的所有数据点以及末尾的颜色条都不一致。 我该如何解决这个问题？

fig, ax = plt.subplots()
for x, y, z in arrays_of_xyz:
     splot = ax.scatter(x.to_numpy(), y.to_numpy(),  marker= 'o', s = z.to_numpy(), cmap ='viridis_r', c = z.to_numpy())

fig.tight_layout()
plt.colorbar(splot)
plt.show()

高瑟姆

Answer 1

看不出 plot 中的尺寸以何种方式不一致。 如果您在调用scatter时不强制执行一致的vmin和vmax ，则颜色条可能会不一致。

您能否尝试使用以下代码并详细说明您遇到的不一致情况：

import numpy as np
import matplotlib.pyplot as plt

num_sets = 3
colors = ("red", "green", "blue")
num_pts_per_set = 20

xs = np.random.randn(num_sets, num_pts_per_set)
ys = np.random.randn(num_sets, num_pts_per_set)
zs = (
    np.random.rand(num_sets, num_pts_per_set)
    * np.arange(1, num_sets + 1).reshape(-1, 1)
    * 30
)
zmin = zs.min()
zmax = zs.max()

fig, (ax1, ax2) = plt.subplots(ncols=2)

ax1.set_title("Sizes according to z\nColors according to set #")
for i, (x, y, z, clr) in enumerate(zip(xs, ys, zs, colors)):
    ax1.scatter(x, y, marker="o", s=z, c=clr, label=f"Set #{i}")
ax1.legend()

ax2.set_title("Facecolors according to z\nSizes according to set #")
for i, (x, y, z, clr) in enumerate(zip(xs, ys, zs, colors)):
    splot = ax2.scatter(x, y, marker="o", c=z, edgecolors=clr, s=(i+1)*30, vmin=zmin, vmax=zmax, label=f"Set #{i}")
ax2.legend()
fig.colorbar(splot)

plt.show()