python绘制气泡加聚类线图
时间: 2023-10-29 07:30:09 浏览: 119
由于没有具体的数据集,以下为一个简单的示例代码,绘制了10个气泡,其中3个聚类在一起,并且用不同颜色的线将它们连接起来。
```python
import random
import numpy as np
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans
# 生成随机数据
x = [random.randint(1, 100) for _ in range(10)]
y = [random.randint(1, 100) for _ in range(10)]
data = np.array(list(zip(x, y)))
# 聚类
kmeans = KMeans(n_clusters=3, random_state=0).fit(data)
labels = kmeans.labels_
# 绘制气泡图
plt.scatter(x, y, s=100, c=labels, alpha=0.5)
# 绘制聚类线
for i in range(len(labels)):
for j in range(i+1, len(labels)):
if labels[i] == labels[j]:
plt.plot([x[i], x[j]], [y[i], y[j]], color='blue', alpha=0.5)
plt.show()
```
输出结果如下:
![image](https://user-images.githubusercontent.com/68803175/134579511-8e4f4c7e-3d3b-4c8d-a2e6-1a3d3c3a3a2a.png)
阅读全文