一维数据聚类python代码
时间: 2023-10-06 19:06:11 浏览: 182
没用的:【微信小程序】资源
以下是一维数据聚类的 Python 代码示例:
```python
import numpy as np
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans
# 生成一维数据
data = np.random.randn(100)
# 聚类数量
k = 3
# KMeans 聚类
kmeans = KMeans(n_clusters=k)
kmeans.fit(data.reshape(-1, 1))
# 聚类结果
labels = kmeans.labels_
# 绘制散点图
plt.scatter(np.arange(len(data)), data, c=labels)
plt.show()
```
代码中使用了 `numpy`、`matplotlib` 和 `sklearn.cluster.KMeans` 模块。首先生成一维数据,然后定义聚类数量 `k`,接着使用 `KMeans` 模型进行聚类,并将聚类结果可视化。
阅读全文