有好几个油井,他们的每天的耗电量不同。python语言用kmeans算法写一个对excel的中油井和耗电量进行分析,进行聚类并画出来。
时间: 2024-02-09 22:11:06 浏览: 87
python中使用k-means聚类.zip_k-means聚类算法_python_python 用kmeans_聚类_聚类 P
好的,下面是完整的Python代码示例,用于对Excel中的油井和耗电量进行聚类,并绘制出聚类结果的散点图:
```python
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans
# 读取Excel文件
df = pd.read_excel('data.xlsx', index_col=0)
# 进行聚类
kmeans = KMeans(n_clusters=3)
kmeans.fit(df)
# 将聚类结果添加到DataFrame对象中
df['cluster'] = kmeans.labels_
# 绘制散点图
colors = ['r', 'g', 'b']
for i in range(3):
plt.scatter(df[df['cluster']==i]['耗电量'], df[df['cluster']==i].index, c=colors[i], label=f'Cluster {i}')
plt.legend()
plt.xlabel('耗电量')
plt.ylabel('油井名称')
plt.show()
```
在此示例中,我们假设Excel文件名为`data.xlsx`,其中第一列为油井名称,第二列为耗电量。我们将数据分为3类,并使用不同的颜色表示不同的聚类结果。你可以根据需要修改聚类数量和颜色方案。
阅读全文