如何遍历DataFrame按parm列唯一值分组导出对应名称的CSV文件
解决方案
以下是基于Pandas实现需求的可直接运行代码:
import pandas as pd # 若数据从本地CSV读取可取消注释下一行替换为你的源文件路径 # df = pd.read_csv("你的源数据路径.csv") # 方式1:自动遍历parm列所有唯一值导出,会包含所有存在的parm分组 for group_name, group_df in df.groupby('parm'): # 导出为对应名称的CSV,index=False用于去掉pandas默认输出的行号索引 group_df.to_csv(f"{group_name}.csv", index=False) # 方式2:仅导出指定的两个分组,可避免示例中最后一行WATSYRI这类拼写异常生成多余文件 target_parms = ['BMP_ASR1', 'WATSYRIN'] for parm_val in target_parms: filtered_df = df[df['parm'] == parm_val] filtered_df.to_csv(f"{parm_val}.csv", index=False)
- 两种方式导出的CSV都会保留原DataFrame的全部原始列,仅包含对应parm值的行
- 若需要保留导出文件的行号索引,删除
to_csv方法中的index=False参数即可
内容的提问来源于stack exchange,提问作者user11958450
相关产品推荐
相关产品推荐

