我有一个带数据的矩阵x,其中每一行都有一个类,一个向量y表示类的类型(可以是0或1)。我想绘制x的散射矩阵,其中每种类型的颜色都有一个颜色(例如绿色表示1的颜色,蓝色表示0的颜色)。我试过
#Define the color for each class
c={0.0:'blue',1.0:'green'}
#Plot it
df = pnd.DataFrame(x,columns=['PC1', 'PC2', 'PC3', 'PC4'])
pnd.scatter_matrix(df, alpha=0.2, figsize=(10, 10), color=c[y], diagonal='None')
也许它没有任何意义,但我是Python的新手,我现在对此并不多。提前谢谢