写代码实现对鸢尾花进行K-means算法聚类,并用matplot画图吗、,数据集是iris_test
时间: 2023-12-03 12:45:51 浏览: 105
好的,下面是代码实现:
```python
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans
from sklearn.datasets import load_iris
# 载入数据集
iris = load_iris()
X = iris.data
y = iris.target
# K-means聚类
kmeans = KMeans(n_clusters=3, random_state=0).fit(X)
# 可视化结果
plt.scatter(X[:, 0], X[:, 1], c=kmeans.labels_)
plt.xlabel('Sepal length')
plt.ylabel('Sepal width')
plt.title('K-means clustering on iris dataset')
plt.show()
```
运行代码后,会出现一张图像,表示聚类结果。其中,不同颜色的点代表不同的类别。
如果你需要对其它特征进行聚类,只需要修改代码中的`X[:, 0]`和`X[:, 1]`即可。
阅读全文