请运用k-means聚类算法将这400名批发商客户进行聚类,给出代码
时间: 2023-07-01 09:27:54 浏览: 133
好的,下面是使用Python实现k-means聚类算法的代码,以聚类成3类为例:
```python
import pandas as pd
from sklearn.cluster import KMeans
from sklearn.preprocessing import StandardScaler
# 读取数据
data = pd.read_csv('wholesale_customers_data.csv')
# 标准化数据
scaler = StandardScaler()
data_scaled = scaler.fit_transform(data)
# 初始化k-means模型
kmeans = KMeans(n_clusters=3, random_state=42)
# 训练模型
kmeans.fit(data_scaled)
# 获取聚类结果标签
labels = kmeans.labels_
# 添加聚类结果标签到原数据
data['Cluster'] = labels
# 输出聚类结果
print(data)
```
其中,'wholesale_customers_data.csv'是数据文件名,需要根据实际情况修改。
需要注意的是,k-means算法的聚类结果对于初始化质心点的选择具有一定的随机性,不同的初始值可能会导致不同的聚类结果。因此,可以尝试多次运行算法,选取最优的聚类结果。
阅读全文