使用PCA要素作为X和Y轴绘制集群

时间:2019-01-21 11:20:26

标签: python python-3.x dataframe matplotlib plotly

我已将PCA应用于数据框,以便基于K均值绘制聚类。由于我喜欢原始df中的24个特征,因此我不想每次仅基于3个或3个特征绘制聚类。因此,我要做的是绘制这些功能的组合,以获得群集中每个功能的更一般/更具代表性的图形表示。

我使用pca.components_提取了组件,并创建了以下df组件:

                         PC-1       PC-2
media_bi_mov         0.003094   0.050599
media_bi_post        0.000762   0.028931
total_mov_prod_300   0.000836   0.573675
codsprod_0           0.440476  -0.004404
codsprod_1           0.008005   0.105349
codsprod_2           0.002851   0.042459
codsprod_3           0.001078   0.009355
codsprod_4          -0.011922  -0.022020
idaplic_0            0.392229  -0.002817
idaplic_1            0.003001   0.004822
idaplic_2            0.044730  -0.001148
idaplic_3            0.097695  -0.008628
idaplic_4            0.024273   0.486973
idaplic_5            0.234798  -0.033369
idaplic_6            0.019329   0.015455
idempro_36           0.000401  -0.000438
idempro_38           0.032149   0.292137
idempro_49           0.439413  -0.023269
codmonsw_EUR         0.440543  -0.002770
codmonsw_USD         0.000378   0.000664
resto_codsprod       0.011406   0.011731
resto_idaplic        0.041649   0.005692
días_entre_ops      -0.011129  -0.015144
frecuencia           0.440543  -0.002770
valor_total_eur      0.000836   0.573675

通常,如果这是原始df,我会使用kmeans.labels_绘制群集以对每个群集应用不同的颜色。但是现在我的问题是,由于kmeans.labels_的长度较大,因此我无法使用kmeans.labels_来区分此经过pca缩减的df中的每个群集。

我该如何应用颜色来区分此数据帧中的聚类?

预先感谢

1 个答案:

答案 0 :(得分:1)

我没有意识到解决这个问题的方法如此简单:我只需要在组件df上运行kmeans来获取每个主要组件中每个功能的集群标签。希望这会帮助和我一样有疑问的人。