Matplotlib散点图如何用长度42的1D数组为42×100的2D坐标数组着色

Matplotlib的scatter函数会自动将传入的二维x、y数组展平为一维，总共有42*100=4200个独立散点，因此要求颜色参数c的长度必须和散点总数一致，这就是你之前报错的根本原因。

你不需要修改原始颜色映射逻辑，只需要将长度为42的颜色数组按行重复100次，让同一行的所有散点共享同一个颜色值即可，完全不影响后续colorbar的展示逻辑：

import numpy as np
import matplotlib.pyplot as plt

x = np.random.rand(42,100)
y = np.random.rand(42,100)
color = np.arange(0,42,1)

# 将颜色数组按行重复，匹配散点总数
c_expanded = np.repeat(color, 100)

plt.scatter(x, y, s=5, c=c_expanded, cmap='jet')
plt.colorbar()
plt.show()

方案说明 #

• np.repeat(color, 100)会将每个颜色值重复100次，最终得到长度为4200的数组，恰好匹配展平后的散点数量
• 重复后的数组数值范围和原始长度42的color数组完全一致，colormap映射规则、colorbar的刻度范围和展示效果和你预期的完全相同，没有任何精度损失

如果需要更灵活的按行自定义配置，也可以逐行绘制散点，配合ScalarMappable生成统一的colorbar：

from matplotlib.cm import ScalarMappable
from matplotlib.colors import Normalize

x = np.random.rand(42,100)
y = np.random.rand(42,100)
color = np.arange(0,42,1)

norm = Normalize(vmin=color.min(), vmax=color.max())
cmap = plt.get_cmap('jet')

for i in range(42):
    plt.scatter(x[i], y[i], s=5, color=cmap(norm(color[i])))

# 手动添加colorbar
plt.colorbar(ScalarMappable(norm=norm, cmap=cmap))
plt.show()

内容的提问来源于stack exchange，提问作者wabash