有條件的顏色與matplotlib散射

Question

我有以下的熊貓數據框，其中列代表一個虛擬變量：

Panda Dataframe

我想要做的就是給我的標記一個cmap='jet'色以下的值b列除外，當列值爲1時，我希望它是灰色的。

任何想法我可以做到這一點？

Answer 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代表熊貓系列。