扫码加入数据分析学习群

问题背景：

随机抽取2个特征，画二维聚类图，提示此TypeError

问题原代码：

n_clusters = 4
color = ["red","pink","orange","gray"]
for i in range(n_clusters):
    plt.scatter(X[y_pred==i,0],X[y_pred==i,1]
               ,marker="x",
               ,s=8
               ,c=color[i]
               )
plt.scatter(centroid[:,0],centroid[:,1]
           ,marker="x"
           ,s=15
           ,c="black");

问题代码报错图片：

解决

将DataFrame对象转成ndarray数组即可

n_clusters = 4
color = ["red","pink","orange","gray"]
for i in range(n_clusters):
    plt.scatter(X[y_pred==i,0],X[y_pred==i,1]
               ,marker="x",
               ,s=8
               ,c=color[i]
               )
plt.scatter(centroid.values[:,0],centroid.values[:,1]
           ,marker="x"
           ,s=15
           ,c="black");