kmeans ++聚类算法python代码
时间: 2023-08-28 21:07:30 浏览: 142
### 回答1:
以下是KMeans++聚类算法的Python代码:
```python
import numpy as np
from sklearn.cluster import KMeans
def kmeans_plus_plus(X, n_clusters):
"""
KMeans++聚类算法
:param X: 数据集
:param n_clusters: 聚类数
:return: 聚类结果
"""
# 随机选择一个样本作为第一个聚类中心
centers = [X[np.random.randint(len(X))]]
# 计算每个样本到最近聚类中心的距离的平方
distances = np.array([np.min([np.linalg.norm(x - c) ** 2 for c in centers]) for x in X])
# 选择剩余的聚类中心
for _ in range(n_clusters - 1):
# 根据距离的平方占比,以概率选择下一个聚类中心
probabilities = distances / np.sum(distances)
centers.append(X[np.random.choice(len(X), p=probabilities)])
distances = np.minimum(distances, np.array([np.linalg.norm(x - c) ** 2 for c in centers]))
# KMeans算法进行聚类
kmeans = KMeans(n_clusters=n_clusters, init=np.array(centers), n_init=1)
kmeans.fit(X)
return kmeans.labels_
```
使用方法:
```python
# 生成随机数据集
X = np.random.randn(100, 2)
# KMeans++聚类
labels = kmeans_plus_plus(X, n_clusters=3)
```
### 回答2:
K-means 是一种经典的聚类算法,它可以将一组数据分成预定的 K 个簇。下面是使用 Python 实现 K-means 算法的示例代码:
```python
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_) # 打印每个样本所属的簇
print(kmeans.cluster_centers_) # 打印簇中心点的坐标
```
在这个例子中,我们使用了 `numpy` 库创建了包含 6 个样本的示例数据集 X。然后,我们使用 `sklearn.cluster` 库中的 KMeans 类构建了一个 K-means 模型,并将模型拟合到数据集 X 上。我们指定了 `n_clusters` 参数为 2,表示我们希望将数据集分成 2 个簇。模型拟合完成后,我们可以通过 `labels_` 属性获取每个样本所属的簇,并通过 `cluster_centers_` 属性获取簇的中心点坐标。
以上就是一个简单的 K-means 算法的 Python 代码示例。请注意,这只是一个基本的示例,实际应用中,我们通常会预处理数据,并根据具体问题调整模型的参数。
### 回答3:
K均值聚类算法是一种常用的无监督学习算法,通过将数据集划分为K个不同的簇来完成聚类任务。下面给出一个简单的K均值聚类算法的Python代码示例:
```python
import numpy as np
def kmeans(X, K, max_iters=100):
# 初始化簇中心
centroids = X[np.random.choice(range(len(X)), K, replace=False)]
for _ in range(max_iters):
# 分配每个样本到最近的簇中心
labels = assign_clusters(X, centroids)
# 更新簇中心
centroids = update_centroids(X, labels, K)
return labels, centroids
def assign_clusters(X, centroids):
labels = []
for x in X:
# 计算样本与各个簇中心的距离
distances = np.linalg.norm(x - centroids, axis=1)
# 将样本分配到距离最近的簇
labels.append(np.argmin(distances))
return np.array(labels)
def update_centroids(X, labels, K):
centroids = []
for i in range(K):
# 获取属于第i个簇的样本
cluster_samples = X[labels == i]
# 计算簇的质心
centroid = np.mean(cluster_samples, axis=0)
centroids.append(centroid)
return np.array(centroids)
# 测试
X = np.array([[1, 2], [2, 1], [3, 4], [4, 3]])
K = 2
labels, centroids = kmeans(X, K)
print("样本分配结果:", labels)
print("簇中心:", centroids)
```
这段代码首先通过随机选择K个样本作为初始的簇中心。在每一次迭代中,首先将每个样本分配到距离最近的簇中心,然后根据新的簇分配更新簇中心。迭代结束后,返回最终的样本分配结果和簇中心。以上是一个简化的K均值聚类算法的代码实现,实际应用中可能还会考虑更多的细节和改进。
阅读全文