python聚类图 颜色
时间: 2023-10-14 13:16:36 浏览: 239
python 聚类 效果图.rar
你可以使用 Python 中的 Matplotlib 库来绘制聚类图,并为每个聚类分配不同的颜色。以下是一个简单的示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans
# 生成随机数据
X = np.random.rand(50, 2)
# 执行聚类
kmeans = KMeans(n_clusters=3, random_state=0).fit(X)
labels = kmeans.labels_
# 绘制聚类结果
colors = ['r', 'g', 'b']
for i in range(len(X)):
plt.scatter(X[i, 0], X[i, 1], color=colors[labels[i]])
plt.show()
```
在这个例子中,我们生成了一个随机数据集并执行了 K-Means 聚类算法。然后,我们为每个聚类分配了一种颜色,并使用 Matplotlib 库中的 scatter 函数绘制了聚类图。每个数据点的颜色对应于其所属的聚类。
阅读全文