当然,以下是一个使用 sklearn 库中的 KMeans 类来实现 K-Means 算法的简单 Python 代码示例。如果你想从头开始实现 K-Means 算法,那将需要更多的代码,但以下是一个使用 sklearn 的示例:
python代码
from sklearn.cluster import KMeans |
|
import numpy as np |
|
import matplotlib.pyplot as plt |
|
# 假设我们有一些二维数据 |
|
data = np.array([[1, 2], [1, 4], [1, 0], |
|
[10, 2], [10, 4], [10, 0]]) |
|
# 设定要聚类的数量(例如,2) |
|
num_clusters = 2 |

最低0.47元/天 解锁文章





