kmeans python代码
时间: 2023-07-09 11:14:13 浏览: 58
以下是KMeans算法的Python代码示例:
```python
from sklearn.cluster import KMeans
import numpy as np
# 构造数据
X = np.array([[1, 2], [1, 4], [1, 0], [4, 2], [4, 4], [4, 0]])
# 定义KMeans模型
kmeans = KMeans(n_clusters=2, random_state=0)
# 训练模型
kmeans.fit(X)
# 打印聚类结果
print(kmeans.labels_)
```
在上面的代码中,我们使用了`scikit-learn`库中的`KMeans`类来实现KMeans算法。我们首先构造了一个二维数据集`X`,然后定义了一个KMeans模型并将其训练。在训练完成后,我们可以通过`labels_`属性来获取每个数据点所属的簇的标签。
相关问题
kmeans python 代码
可以参考以下代码实现kmeans聚类:
```python
from sklearn.cluster import KMeans
import numpy as np
# 载入数据
data = np.loadtxt('data.txt')
# 定义KMeans模型
kmeans = KMeans(n_clusters=3)
# 训练模型
kmeans.fit(data)
# 输出聚类结果
print(kmeans.labels_)
```
其中,`data.txt`是待聚类的数据文件,需要自行准备。`n_clusters`定义聚类数目。`kmeans.labels_`输出了每个样本所属的类别。
kmeans python实现代码
当然可以,以下是一个简单的kmeans算法的Python实现代码:
```
import numpy as np
def kmeans(X, K, max_iters=100):
# 随机初始化聚类中心
centroids = X[np.random.choice(range(len(X)), K, replace=False)]
for i in range(max_iters):
# 计算每个样本所属的聚类中心
distances = np.sqrt(((X - centroids[:, np.newaxis])**2).sum(axis=2))
cluster = np.argmin(distances, axis=0)
# 更新聚类中心
for k in range(K):
centroids[k] = X[cluster==k].mean(axis=0)
return cluster, centroids
```
希望能帮到你!
阅读全文