当然,以下是一个使用 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 |
|
# 使用 sklearn 的 KMeans 类 |
|
kmean |