帮我写一段k-mean聚类分析的python代码
时间: 2024-03-02 11:27:53 浏览: 106
K-means聚类python代码
下面是一个k-mean聚类分析的Python代码示例:
```
import numpy as np
from sklearn.cluster import KMeans
# 创建数据集
X = np.array([[1, 2], [1, 4], [1, 0], [4, 2], [4, 4], [4, 0]])
# 创建KMeans对象
kmeans = KMeans(n_clusters=2, random_state=0)
# 对数据进行聚类
kmeans.fit(X)
# 输出聚类结果
print(kmeans.labels_)
```
在这个例子中,我们首先创建了一个包含6个点的数据集X,然后创建了一个KMeans对象,并将其设置为2个聚类。接下来,我们使用fit()方法对数据进行聚类,并输出聚类结果。在这个例子中,输出的结果是一个包含6个元素的数组,表示每个点所属的聚类编号。
阅读全文