如何使用KMeans绘制质心图

Question

我正在处理一个数据集，并试图学习如何使用集群分析和KMean。我一开始用散点图绘制两个属性，当我添加第三个属性，并尝试绘制另一个质心时，我得到了一个错误。我正在运行的代码如下：

import numpy as np ##Import necassary packages
import pandas as pd
import matplotlib.pyplot as plt
from matplotlib import style
style.use("ggplot")
from pandas.plotting import scatter_matrix
from sklearn.preprocessing import *
from sklearn.cluster import MiniBatchKMeans 

url2="http://archive.ics.uci.edu/ml/machine-learning-databases/adult/adult.data" #Reading in Data from a freely and easily available source on the internet
Adult = pd.read_csv(url2, header=None, skipinitialspace=True) #Decoding data by removing extra spaces in cplumns with skipinitialspace=True
##Assigning reasonable column names to the dataframe
Adult.columns = ["age","workclass","fnlwgt","education","educationnum","maritalstatus","occupation",  
"relationship","race","sex","capitalgain","capitalloss","hoursperweek","nativecountry",
"less50kmoreeq50kn"]
Adult.loc[:, "White"] = (Adult.loc[:, "race"] == "White").astype(int)
X = pd.DataFrame()
X.loc[:,0] = Adult.loc[:,'age']
X.loc[:,1] = Adult.loc[:,'hoursperweek']
X.loc[:,2] = Adult.loc[:, "White"]
kmeans = MiniBatchKMeans(n_clusters = 3)
kmeans.fit(X)
centroids = kmeans.cluster_centers_
labels = kmeans.labels_
print(centroids)
print(labels)
colors = ["green","red","blue"]
plt.scatter(X.iloc[:,0], X.iloc[:,1], X.iloc[:,2], c=np.array(colors)[labels], alpha=.1)
plt.scatter(centroids[:, 0], centroids[:, 1],  marker = "x", s=150, 
linewidths = 5, zorder = 10, c=['green', 'red','blue'])
plt.show()

运行该代码是可行的，但它似乎不正确，因为只有2个质心被"调用"，但仍有3个质心被绘制。当我将质心散点图更改为：时

plt.scatter(centroids[:, 0], centroids[:, 1], centroids[:, 2] marker = "x", s=150, 
linewidths = 5, zorder = 10, c=['green', 'red','blue'])

我得到了TypeError: scatter() got multiple values for argument 's'。原始代码是否不正确？它会在未来的项目中引发问题吗？如果是这样，我应该如何将代码更改为不会收到错误的位置？提前感谢

Answer 1

问题是如果您传递不带关键字的参数值，散点函数希望第三个参数是s。在您的情况下，第三个自变量是质心，您再次将s作为关键字自变量传递。因此，它为s获得了多个值。您需要的是这样的东西。

1( 指定质心列：centroids_x、centroid_sy

centroids_x = centroids[:,0]
centroids_y = centroids[:,1]

2( 制作质心_ x和质心_ y 的散点图

plt.scatter(centroids_x,centroids_y,marker = "x", s=150,linewidths = 5, zorder = 10, c=['green', 'red','blue'])