带有matplotlib散射的条件颜色

Question

我有以下Pandas Dataframe，其中a列代表虚拟变量：

Panda Dataframe

我想做的是在列b的值后面给我的标记cmap='jet'颜色，除非列值为1，我希望它为灰色。

知道我该怎么做吗？

Answer 1

您必须将您的值标记为等于1并绘制：

import matplotlib.pyplot as plt
import numpy as np

# test data
t = np.linspace(0, 2 * np.pi, 30)
x = np.sin(t)
x[3] = 1
y = np.cos(t)

# indices for 'bad' values
indices = x == 1
# calc colors from jet cmap
cmap = plt.get_cmap('jet')
colors = cmap((y - y.min()) / y.ptp())

# normal values
plt.scatter(t[~indices], x[~indices], c = colors[~indices], cmap = cmap)
# bad values
plt.scatter(t[indices], x[indices], c = 'grey')
plt.show()

数组t，x，y代表pandas系列。