PCA 无法在散点图上获得颜色

Question

我正在自己做一个小项目。我正在用 PCA 尝试这个东西。绘制图表后，我似乎无法将颜色显示出来。这些是我的代码的以下步骤。在此之前，我已经对数据进行了缩放和转换。缩放后我做了一个 PCA。这些是步骤。第一：

from sklearn.decomposition import PCA
pca= PCA(n_components= 2)
pca.fit(scaled_data)

x_pca= pca.transform(scaled_data)

principaldf= pd.DataFrame(data=x_pca, 
                      columns=['principal component 1', 
                              'principal component 2'])

在此之后，我结合了两个数据框并得到了这个。

new_df= principaldf.join(df_features)
new_df.head()

接下来我尝试使用以下代码绘制图形：

color= ['r','g']
plt.scatter(x_pca[:, 0], x_pca[:, 1],
     edgecolor='none', alpha=0.5, c= color)
plt.xlabel('component 1')
plt.ylabel('component 2')

我收到了这个错误

ValueError: 'c' 参数有 2 个元素，这是不可接受的 与大小为 261 的“x”一起使用，与大小为 261 的“y”一起使用。

任何人都可以提供建议或帮助吗？希望我的问题足够清楚。谢谢！

Answer 1

你可以试试这样的：

给Gender赋值：

new_df['Gender'] = new_df['Gender'].replace({'Male':0, 'Female':1})

然后使用颜色和 cmap 进行绘图：

plt.scatter(x_pca[:, 0], x_pca[:, 1], edgecolor='none', alpha=0.5,
            c=new_df['Gender'], cmap='RdYlGn')

当您传入像 ['r', 'g'] 这样的 2 项列表时，它不知道哪些点应该是哪种颜色