编写python代码 利用sklearn中的方法对鸢尾花数据集实现k-means算法
时间: 2024-03-14 17:47:06 浏览: 75
K_means_iris_dataset.zip_iris_k-means_kmeans clustering
好的,下面是对鸢尾花数据集实现k-means算法的Python代码:
```python
from sklearn.datasets import load_iris
from sklearn.cluster import KMeans
# 加载鸢尾花数据集
iris = load_iris()
X = iris.data
# 指定聚类的个数
k = 3
# 创建KMeans对象,并进行聚类
kmeans = KMeans(n_clusters=k)
kmeans.fit(X)
# 输出聚类的结果
print("聚类中心点:\n", kmeans.cluster_centers_)
print("聚类结果:\n", kmeans.labels_)
```
在上述代码中,首先我们加载了sklearn中自带的鸢尾花数据集,然后指定了聚类的个数为3。接着,我们创建了一个KMeans对象,并调用fit方法进行聚类。最后,我们通过输出聚类中心点和聚类结果来查看聚类的效果。
需要注意的是,这只是一个简单的示例代码,实际使用中可能需要根据具体情况进行调参等操作。
阅读全文