python - 将 bool 值映射到字符串

Question

我正在绘制一个图表，其中我的 x 变量是“Mg”，我的 y 变量是“Si”。我有第三个变量叫做“二进制”。如果二进制等于 0 或 1，如何分别将标绘点着色为红色或黑色？

我需要使用函数 plt.scatter 和 colourbar()。我读过有关 colourbar 的内容，但它似乎会产生连续的色谱。我试过使用 plt.colors.from_levels_and_colors 代替，但我不确定如何正确使用它。

levels = [0,1]
colors = ['r','b']
cmap, norm = plt.colors.from_levels_and_colors(levels, colors)
plt.scatter(data_train['Mg'], data_train['Si'], c = data_train['binary'])
plt.show()

还有，以后我不在这个论坛上问这样的问题，我该怎么做才能自己解决问题呢？我尝试先在线阅读文档，但经常发现很难理解。

Answer 1

np.where 使编码二进制值变得容易。

np.where([1, 0, 0, 1], 'yes', 'no')
# array(['yes', 'no', 'no', 'yes'], dtype='<U3')

---

colors = np.where(data_train['binary'], 'black', 'red')
plt.scatter(data_train['Mg'], data_train['Si'], c=colors)