python进行k-means聚类分析代码
时间: 2023-09-26 14:09:53 浏览: 110
当进行K-means聚类分析时,您可以使用Python中的`sklearn`库来实现。下面是一个示例代码:
```python
from sklearn.cluster import KMeans
import numpy as np
# 创建示例数据
X = np.array([[1, 2], [1, 4], [1, 0], [4, 2], [4, 4], [4, 0]])
# 创建K-means模型并进行拟合
kmeans = KMeans(n_clusters=2)
kmeans.fit(X)
# 获取聚类结果
labels = kmeans.labels_
centroids = kmeans.cluster_centers_
# 打印聚类结果
for i in range(len(X)):
print("数据点", X[i], "属于聚类", labels[i])
# 打印聚类中心点坐标
print("聚类中心点坐标:")
for i in range(len(centroids)):
print("聚类", i, "中心点:", centroids[i])
```
上述代码中,我们创建了一个包含6个数据点的示例数据集`X`。然后,我们通过实例化`KMeans`对象并指定`n_clusters`参数为2来创建K-means模型。接下来,我们使用`fit()`函数对模型进行拟合,这将根据数据集进行聚类。
然后,我们可以使用`labels_`属性获取每个数据点的所属聚类标签,使用`cluster_centers_`属性获取每个聚类的中心点坐标。
最后,我们打印每个数据点所属的聚类以及每个聚类的中心点坐标。
请根据您的实际数据进行相应的修改,并根据需要调整`n_clusters`参数。希望对您有所帮助!
阅读全文