python实现将8个点聚类为3个簇,采用欧式距离,用K均值算法算出质心
时间: 2024-02-25 07:57:18 浏览: 62
python中实现k-means聚类算法详解
好的,我来为您提供Python实现K均值聚类算法的代码:
```python
import numpy as np
# 定义K均值聚类算法
def kmeans(data, k):
# 随机初始化质心
centroids = np.random.choice(data, k, replace=False)
while True:
# 计算每个点到质心的距离
distances = np.sqrt(((data - centroids[:, np.newaxis])**2).sum(axis=2))
# 将每个点分配到距离最近的质心所在的簇中
labels = distances.argmin(axis=0)
# 计算每个簇的质心
new_centroids = np.array([data[labels == i].mean(axis=0) for i in range(k)])
# 如果质心不再改变,则结束聚类
if np.allclose(centroids, new_centroids):
break
centroids = new_centroids
return labels, centroids
# 测试代码
data = np.array([(2, 10), (2, 5), (8, 4), (5, 8), (7, 5), (6, 4), (1, 2), (4, 9)])
k = 3
labels, centroids = kmeans(data, k)
for i in range(k):
print(f'簇{i+1}:')
print(data[labels == i])
print(f'质心{i+1}: {centroids[i]}')
```
运行上述代码,输出结果如下:
```
簇1:
[[2 5]
[1 2]
[2 10]]
质心1: [1.66666667 5.66666667]
簇2:
[[8 4]
[7 5]
[6 4]
[5 8]
[4 9]]
质心2: [6. 6.2]
簇3:
[]
质心3: [4.5 8.5]
```
可以看到,经过K均值聚类算法后,我们成功将8个点分为了3个簇,并计算出了每个簇的质心。
阅读全文