2016-03-24 19 views
2

elle nasıl renge dizeleri ayarlamak için?Dizgi, python matplotlib dosyasında elle renk nasıl ayarlanır? matplotlib arsa rakamları kullanılırken

Örneğin, şu veri kümesi var.

# Generate data... 
import pandas as pd 
seed(1) 
df = pd.DataFrame({ 
     'x':np.random.random(10), 
     'y':np.random.random(10), 
     'z':np.random.choice(['yes','no'], 10) 
    }) 

df 
      x   y z 
0 0.417022 0.419195 no 
1 0.720324 0.685220 no 
2 0.000114 0.204452 no 
3 0.302333 0.878117 no 
4 0.146756 0.027388 no 
5 0.092339 0.670468 no 
6 0.186260 0.417305 yes 
7 0.345561 0.558690 no 
8 0.396767 0.140387 no 
9 0.538817 0.198101 yes 

Ben 'kırmızı' ve 'evet' çizmek istiyorum 'hayır' kod aşağıdaki ile 'mavi', ama nasıl renkleri ayarlayabilirsiniz olarak?

# Plot... 
plt.scatter(df.x, df.y, c= df.z) 

cevap

2

Bu hat Sen istediğinizi yapacak düşünüyorum:

plt.scatter(df.x, df.y, c=['red' if x == 'yes' else 'blue' for x in df.z])