import numpy as np
import pandas as pd
from matplotlib import pyplot as plt
from sklearn.cluster import KMeans
from sklearn.datasets import load_iris
from sklearn.metrics import silhouette_score
# 导入iris数据
iris=load_iris()# 实例化KMEANS
result=KMeans(n_clusters=3)
result.fit(iris.data)# 预测聚类结果
predict=result.predict(iris.data)# 散点图挑选变量查看聚类情况
plt.figure(figsize=(5,5),dpi=80)
color=['orange','green','blue']
colord=[color[i]for i in predict]
plt.scatter(iris.data[:,2],iris.data[:,3],color=colord)# 轮廓系数评估
silhouette_score(iris.data,predict)