Pandas散射矩阵中的颜色类

2024-05-16 14:22:51 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个包含数据的矩阵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新手,现在对它不太了解。先谢谢你


Tags: the数据类型df颜色绘制代表矩阵