数据挖掘题目，关于聚类算法的题目

from sklearn.cluster import KMeans
from sklearn.datasets import make_blobs
import numpy as np
import matplotlib.pyplot as plt

x_train = np.array([[2,1],[1,2],[2,2],[3,2],[2,3],[3,3],[2,4],[3,5],[4,4],[5,3]])

# k=2
kmeans = KMeans(n_clusters=2)
kmeans.fit(x_train)
y_train = kmeans.predict(x_train)
plt.scatter(x_train[:,0],x_train[:,1],c=y_train) # 预测结果

 输出：

# k=3
kmeans = KMeans(n_clusters=3)
kmeans.fit(x_train)
y_train = kmeans.predict(x_train)
plt.scatter(x_train[:,0],x_train[:,1],c=y_train) # 预测结果

您好，我是有问必答小助手，您的问题已经有小伙伴解答了，您看下是否解决，可以追评进行沟通哦~

如果有您比较满意的答案 / 帮您提供解决思路的答案，可以点击【采纳】按钮，给回答的小伙伴一些鼓励哦～～

ps: 问答会员年卡【8折】购 ，限时加赠IT实体书，即可 享受50次 有问必答服务，了解详情>>>https://t.csdnimg.cn/RW5m