python 聚类图 改颜色
时间: 2023-10-31 13:05:35 浏览: 160
图形颜色修改
3星 · 编辑精心推荐
要改变 Python 中聚类图的颜色,可以使用 Matplotlib 库中的颜色映射(colormap)。以下是一个简单的示例,展示如何在聚类图中使用不同的颜色:
```python
import matplotlib.pyplot as plt
import numpy as np
from sklearn.cluster import KMeans
# 生成一些随机数据
X = np.random.rand(100, 2)
# 进行聚类
kmeans = KMeans(n_clusters=3)
kmeans.fit(X)
y_kmeans = kmeans.predict(X)
# 使用颜色映射来为不同的簇分配不同的颜色
plt.scatter(X[:, 0], X[:, 1], c=y_kmeans, cmap='viridis')
plt.show()
```
在上面的代码中,我们使用 `c` 参数来指定颜色,它会接受一个数组,该数组将为每个数据点指定颜色。我们还通过 `cmap` 参数指定了使用的颜色映射,这里我们使用了 `viridis` 颜色映射。你可以尝试使用其他可用的颜色映射,或者自己创建自定义颜色映射来适应你的需求。
阅读全文