用python写一个k-means数据挖掘代码
时间: 2024-05-02 16:19:59 浏览: 135
数据挖掘中的K-means
好的,以下是一个简单的 Python K-Means 数据挖掘代码:
```
import numpy as np
from sklearn.cluster import KMeans
# 准备数据
X = np.array([[1, 2], [1, 4], [1, 0], [4, 2], [4, 4], [4, 0]])
# 使用 K-Means 进行聚类
kmeans = KMeans(n_clusters=2, random_state=0).fit(X)
# 打印聚类结果
print(kmeans.labels_)
```
这个代码可以将数据 X 分成两个簇。当然,你也可以根据自己的需要修改代码中的参数来实现更复杂的聚类。
阅读全文