如何在 matplotlib 散点图中标准化颜色图？答案

【问题标题】：How can I normalize colormap in matplotlib scatter plot?如何在 matplotlib 散点图中标准化颜色图？
【发布时间】：2017-08-25 05:32:18
【问题描述】：

matplotlib documentation 详细解释了如何规范化 pcolormesh 的颜色图，但我如何才能正确地为散点图做到这一点？

normalize = mcolors.Normalize(vmin=-1, vmax=1)
plt.scatter(x,y,z,cmap=colormap(normalize),marker='*',s=5)

不起作用 (TypeError: Cannot cast array data from dtype('O') to dtype('int64') according to the rule 'safe')

只是 z 数据不完全是从 -1 到 1，我正在绘制多个数据集，其限制在 +/- 0.93 - 98 左右，但我希望颜色以零为中心并从 -1到 1，以便我对所有各种数据集都有相同的参考。

哦，当我不尝试标准化时，我得到TypeError: scatter() got multiple values for keyword argument 's'。显然我不知道如何在散点图中使用颜色图。

【问题讨论】：

标签： python matplotlib colormap

【解决方案1】：

您使用的语法与链接文档中的语法完全不同。归一化散点图或 pcolor(mesh) 或任何其他标量可映射对象之间基本上没有区别。

总是这样

colormap = plt.cm.bwr #or any other colormap
normalize = matplotlib.colors.Normalize(vmin=-1, vmax=1)
plt.scatter(x, y, c=z, s=5, cmap=colormap, norm=normalize, marker='*')

【讨论】：

即使我遵循这个建议，我仍然会得到TypeError: scatter() got multiple values for keyword argument 's' ...所以我建议使用plt.scatter(x,y,c=z,s=5, ...)（可能版本之间的默认值有所变化）
@gluuke 是的，确实，感谢您指出这一点。我更正了答案。

【解决方案2】：

# define the range of values in the data (100 is arbitrary)
my_range = np.linspace(-1,1,100)

# cmap is a function. It returns (rgba) colors base on a range 0-1. Therefore, 
# transform your values to 0-1 to use them as input for cmap() 
cmap = cm.get_cmap('viridis', 100)
my_transformed_range = (my_range - np.min(my_range)) / (np.max(my_range) - np.min(my_range))

# colors should be supplied as a single color or a list (here using the cmap fx)
plt.scatter(my_range, np.ones(100), color=[cmap(i) for i in my_transformed_range])

【讨论】：