在Scikit-learn中,可以通过使用KMeans类来实现K均值聚类。下面是一个简单的示例代码:
from sklearn.cluster import KMeans
import numpy as np
# 创建一些数据用于聚类
X = np.array([[1, 2], [5, 8], [1.5, 1.8], [8, 8], [1, 0.6], [9, 11]])
# 创建KMeans模型并指定聚类的类别数
kmeans = KMeans(n_clusters=2)
# 拟合数据并进行聚类
kmeans.fit(X)
# 获取聚类结果并打印
labels = kmeans.labels_
print(labels)
在上面的代码中,首先创建了一些用于聚类的数据X
,然后创建了一个KMeans
对象,并指定了聚类的类别数为2。接着,使用fit
方法拟合数据并进行聚类,最后获取聚类结果并打印出来。
除了上面的示例代码,Scikit-learn还提供了更多参数和方法,可以根据具体需求来调整模型的设置和使用其他功能。更多关于K均值聚类的信息可以参考Scikit-learn的官方文档:https://scikit-learn.org/stable/modules/generated/sklearn.cluster.KMeans.html