如何用Pandas打印DataFrame单列到CSV及解决列访问报错

一、列访问KeyError问题解决 #

原因分析 #

从你提供的DataFrame输出看，列名之间存在多空格分隔，大概率是加载CSV时未正确解析分隔符，导致列名被错误识别（比如包含前后空格、或者列名未被正确拆分），进而无法通过你输入的列名正常访问。

解决步骤 #

1. 查看真实列名
   先打印所有列的真实名称，确认是否有隐藏空格或格式问题：

   print(df.columns.tolist())
   

2. 修复列名或加载方式
   有两种方案可选：

   • 方案1：重新正确加载CSV
     若原CSV采用空格/制表符分隔，加载时指定分隔符：

     df = pd.read_csv('cali_avocados.csv', sep='\s+')
     

     这样会自动拆分多空格分隔的列，列名会被正确识别。

   • 方案2：清理现有列名
     对已加载的DataFrame，清理列名的前后空格，并将中间空格替换为下划线（方便后续访问）：

     # 去除列名前后空格，替换中间空格为下划线
     df.columns = df.columns.str.strip().str.replace(' ', '_')
     

     处理后即可用df['Production']这类简洁名称访问列。

二、将DataFrame单列导出到CSV #

直接选中目标列后调用to_csv方法，添加index=False可避免导出多余的行索引：

# 示例：导出Production列到production.csv
df['Production'].to_csv('production.csv', index=False)

如果需要保留行索引，去掉index=False参数即可。

内容的提问来源于stack exchange，提问作者coja56