鸢尾花数据集聚类结果可视化

该代码使用matplotlib和pandas对iris_cluster.csv数据集进行处理,将不同标签的鸢尾花数据点用不同颜色在二维平面上表示出来,进行了聚类后的可视化展示,分别用蓝色圆圈(bo)、红色星号(*)和绿色正方形(gD)代表三种鸢尾花类别。
摘要由CSDN通过智能技术生成

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
plt.rcParams["font.sans-serif"] = "SimHei"
plt.rcParams["axes.unicode_minus"] = False

df = pd.read_csv('iris_cluster.csv', index_col=0)
ls_labels = sorted(np.unique(df['labels']))
df1 = df[df['labels'] == ls_labels[0]]
df2 = df[df['labels'] == ls_labels[1]] 
df3 = df[df['labels'] == ls_labels[2]]  
fig = plt.figure(figsize=(9,6),dpi=200)                    ##设定空白画布为9*6,分辨率为200
plt.title("聚类二维可视化")                          #设置标题
plt.plot(df1.iloc[:,0],df1.iloc[:,1],'bo')
plt.plot(df2.iloc[:,0],df2.iloc[:,1],'r*')
plt.plot(df3.iloc[:,0],df3.iloc[:,1],'gD')
plt.legend(["setosa","versicolor","virginica"])                         #设置图例
plt.savefig('iris_cluster.png')

好的,我可以为您介绍一下如何使用Kmeans算法来实现鸢尾花数据集聚类,并进行可视化展示。 首先,我们需要导入必要的库,包括numpy,pandas,matplotlib和sklearn等。您可以使用以下代码来导入这些库: ```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 ``` 然后,我们可以使用load_iris函数来加载鸢尾花数据集。这个函数将返回一个Bunch对象,其中包含数据集的特征和标签等信息。我们可以使用以下代码来加载数据集: ```python data = load_iris() X = data.data y = data.target ``` 接下来,我们可以使用KMeans算法来对数据集进行聚类。KMeans是一种无监督的学习算法,它可以将数据集中的样本分成指定数量的簇。我们可以使用以下代码来创建一个KMeans对象,并使用fit方法来训练模型: ```python kmeans = KMeans(n_clusters=3, random_state=0) kmeans.fit(X) ``` 在训练完成后,我们可以使用predict方法来预测每个样本所属的簇。然后,我们可以使用scatter函数来将预测结果可视化展示。我们可以使用以下代码来实现: ```python plt.scatter(X[:, 0], X[:, 1], c=kmeans.labels_) plt.xlabel('Sepal length') plt.ylabel('Sepal width') plt.title('KMeans clustering') plt.show() ``` 这将会展示一个散点图,其中每个点的颜色表示它所属的簇。您可以通过修改n_clusters参数的值来改变簇的数量,从而观察不同数量的簇对聚类效果的影响。 完整代码如下: ```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 data = load_iris() X = data.data y = data.target kmeans = KMeans(n_clusters=3, random_state=0) kmeans.fit(X) plt.scatter(X[:, 0], X[:, 1], c=kmeans.labels_) plt.xlabel('Sepal length') plt.ylabel('Sepal width') plt.title('KMeans clustering') plt.show() ``` 希望这能对您有所帮助!
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值