使用kmeans对鸢尾花数据集聚类

海边菠萝

已于 2023-07-19 16:55:05 修改

阅读量791

点赞数 2

文章标签： python 分类

于 2023-07-19 16:52:04 首次发布

本文链接：https://blog.csdn.net/m0_63054020/article/details/131812986

版权

#使用kmeans对鸢尾花数据集聚类
from sklearn.datasets import load_iris
from sklearn.cluster import KMeans
import matplotlib.pyplot as plt
import numpy as np
import matplotlib
from sklearn.model_selection import train_test_split 
matplotlib.rcParams['font.family']='YouYuan'

#加载数据
iris=load_iris()
data=iris.data
target=iris.target
x=data[:,[0,2]] #行数不限，列取第1，3列，列是从0开始算的，所以不是1，3而是0，2
y=iris.target
print(y)
label=np.array(y)
index_0=np.where(label==0)
plt.scatter(x[index_0,0],x[index_0,1],marker='o',color='g',label='0')

index_1=np.where(label==1)
plt.scatter(x[index_1,0],x[index_1,1],marker='*',color='b',label='1')

index_2=np.where(label==2)
plt.scatter(x[index_2,0],x[index_2,1],marker='*',color='r',label='2')

plt.xlabel('花萼长度')
pl