Python中如何将KMeans聚类输出的kmc坐标列表导出为CSV文件
你可以直接借助你已经在使用的pandas库完成导出,操作如下:
- 先将
km.cluster_centers_返回的numpy数组转换为DataFrame格式,可同时自定义列名方便后续使用 - 调用DataFrame自带的
to_csv方法导出文件即可
示例代码:
# 转换质心数组为DataFrame,指定X、Y为列名 centroid_df = pd.DataFrame(kmc, columns=["X", "Y"]) # 导出为CSV文件 # index=False 可以避免导出多余的序号列 # encoding='utf-8-sig' 是为了保证用Excel打开文件时不会出现乱码,不需要用Excel打开可改为utf-8 centroid_df.to_csv("kmeans_centroid_output.csv", index=False, encoding="utf-8-sig")
如果需要控制导出的坐标小数点精度,可以增加float_format参数,比如保留8位小数的写法为:
centroid_df.to_csv("kmeans_centroid_output.csv", index=False, encoding="utf-8-sig", float_format="%.8f")
内容的提问来源于stack exchange,提问作者user17335922
相关产品推荐
相关产品推荐

