2016-04-13 159 views
0

我有一個有3個特徵(列)一個熊貓據幀,即X,Y和Z X和Y是包含數據列表,而Z含有字符串值的列表(顏色 - 紅色&藍色)。我想與x軸作爲特徵X和y軸與特徵量y的散點圖,對於根據在Z.更改顏色的功能

對於離值有色散射點。如果一行中的Z值爲紅色,則將該點着色爲紅色,如果其藍色則爲藍色。

data_frame['X'] = pandas.Series(X) 
data_frame['Y'] = pandas.Series(Y) 
data_frame['Z'] = pandas.Series(Z) 

data_frame.plot(kind='scatter', x='X', y='Y') 
pl.show() 

情節工作正常,但我無法弄清楚如何實現顏色部分。有任何想法嗎??

回答

1
import numpy as np 
import matplotlib.pyplot as plt 
N = 50 
x = np.random.rand(N)  # pandas series works too 
y = np.random.rand(N) 
colors = np.random.rand(N) 
circle = 100* np.ones(50) 
plt.scatter(x, y, c=colors, s=circle, alpha=0.6) 
plt.show() 

the resulting plot

+0

作品!謝謝!! – Krishh

2

你可以嘗試使用seaborn:

import seaborn as sns 
sns.lmplot("X", "Y", data= data_frame, hue='Z', fit_reg=False) 
+0

它做工精細,但你能告訴我怎麼把它與任何matplotlib或熊貓嗎?我想盡可能少使用模塊。 – Krishh