创建图表时出现未知索引错误答案

【问题标题】：Unknown IndexError while creating graph创建图表时出现未知索引错误
【发布时间】：2020-01-02 19:28:56
【问题描述】：

我试图在使用 PCA 和 k 之后输出一个集群图 - 表示使用以下代码进行集群：

Mergory=Mergory.astype(np.float32)
scaler = StandardScaler()
scaler.fit(Mergory)
Mergory = scaler.transform(Mergory)
pca.fit(Mergory)
Mergory = pca.transform(Mergory)

from sklearn.cluster import KMeans
kmeans = KMeans(n_clusters=3)
kmeans.fit(Mergory)
labels = kmeans.predict(Mergory)
centroids = kmeans.cluster_centers_

然后我尝试使用我在网上找到的以下编码以图表形式获取输出：

fig = plt.figure(figsize=(5, 5))

colors = map(lambda x: colmap[x+1], labels)

plt.scatter(Mergory['hospitalizations_one_year'], Mergory['fried'], color=colors, alpha=0.5, edgecolor='k')
for idx, centroid in enumerate(centroids):
    plt.scatter(*centroid, color=colmap[idx+1])
plt.show()

我得到这个错误：

IndexError                                Traceback (most recent call last)
<ipython-input-26-da3a8ac504f4> in <module>
      3 colors = map(lambda x: colmap[x+1], labels)
      4 
----> 5 plt.scatter(Mergory['hospitalizations_one_year'], Mergory['fried'], color=colors, alpha=0.5, edgecolor='k')
      6 for idx, centroid in enumerate(centroids):
      7     plt.scatter(*centroid, color=colmap[idx+1])

IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices

<Figure size 360x360 with 0 Axes>

有谁知道解决这个问题或者是什么原因造成的？

【问题讨论】：

标签： python matplotlib scikit-learn

【解决方案1】：

您在那里将pd.DataFrame 转换为np.array：

Mergory = scaler.transform(Mergory)
pca.fit(Mergory)
Mergory = pca.transform(Mergory)

因此，您不能像在此处那样使用列索引：

Mergory['hospitalizations_one_year'], Mergory['fried']

这个例子抛出了同样的错误：

import numpy as np
arr = np.random.rand(4, 4)
arr['col 1']

IndexError：只有整数、切片 (:)、省略号 (...)、numpy.newaxis (None) 和整数或布尔数组是有效的索引

【讨论】：