python读取excel文档使用kmeans聚类算法绘制散点图
以下是使用Python读取Excel文档并使用K-means聚类算法绘制散点图的示例代码:
import pandas as pd
import numpy as np
from sklearn.cluster import KMeans
import matplotlib.pyplot as plt
# 读取Excel文档
df = pd.read_excel('data.xlsx')
# 提取需要聚类的数据
data = df.iloc[:, [2, 3]].values
# 使用K-means算法进行聚类
kmeans = KMeans(n_clusters=3, init='random')
kmeans.fit(data)
labels = kmeans.labels_
# 绘制散点图
colors = ['r', 'g', 'b']
for i in range(len(data)):
plt.scatter(data[i][0], data[i][1], c=colors[labels[i]])
# 显示图形
plt.show()
在这个示例中,我们首先使用Pandas库读取Excel文档,并从中提取需要聚类的数据。然后,我们使用scikit-learn库中的K-means算法进行聚类,并将聚类结果保存在labels变量中。最后,我们使用Matplotlib库绘制散点图,其中不同聚类的数据点使用不同的颜色表示
原文地址: https://www.cveoy.top/t/topic/hhua 著作权归作者所有。请勿转载和采集!