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')