python分别保存聚类分析结果+KeyError: ‘CustomerID‘报错

如何在完成聚类分析后按聚类编号保存数据并且带上原数据所属ID

# 将每个聚类的数据保存到不同的文件中
for cluster_id in range(6):  # 假设共有6个聚类
    cluster_data = data[data['cluster'] == cluster_id]
    cluster_data_with_customer_id = cluster_data.copy()
    cluster_data_with_customer_id['CustomerID'] = data.loc[cluster_data.index, 'CustomerID']
    cluster_data_with_customer_id.to_excel(f'cluster_{cluster_id}_data.xlsx', index=False)  # 将数据保存为Excel文件,文件名包含聚类编号,并包含CustomerID列

在保存聚类分析结果时我希望带上每条数据原来所属的CustomerID方便后续添加对应的数据进行分析,结果报错KeyError: 'CustomerID'。

这是因为我读取原数据中没有包含CustomerID列,导致dataframe里面没有CustomerID。

解决:

在读取数据的时候加入CustomerID列

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值