批量为CSV文件添加缺失列q17/q20的Python脚本报错求助

你这段脚本的核心问题有两个：

• 把文件路径列表onlyfiles直接赋值给了df，这是个普通Python列表，不是pandas的DataFrame对象，自然没有DataFrame的列操作能力，所以触发了“‘list’ object has no attribute columns”错误（实际运行还会触发列表不能用df['q17']赋值的错误）。
• 就算拿到了正确的DataFrame，判断列是否存在的写法if ['q17'] not in df也是错误的，应该检查列名是否在df.columns集合里。

import pandas as pd
from os import listdir
from os.path import isfile, join

# 获取当前目录下的所有CSV文件（只处理CSV，避免误操作其他文件）
onlyfiles = [f for f in listdir('.') if isfile(join('.', f)) and f.endswith('.csv')]

# 遍历每个CSV文件批量处理
for file in onlyfiles:
    # 读取CSV文件为pandas DataFrame对象
    df = pd.read_csv(file)
    
    # 检查q17列是否存在，缺失则添加空值列
    if 'q17' not in df.columns:
        df['q17'] = ''
    
    # 检查q20列是否存在，缺失则添加空值列
    if 'q20' not in df.columns:
        df['q20'] = ''
    
    # 保存修改后的文件（index=False避免自动添加索引列）
    df.to_csv(file, index=False)
    print(f"处理完成：{file}")

• 新增f.endswith('.csv')过滤条件，确保只处理CSV文件，避免误操作其他格式文件。
• 遍历每个文件时先读取为DataFrame，再执行列检查与添加操作，符合pandas的操作逻辑。
• 若不想覆盖原文件，可修改保存路径为f"processed_{file}"，生成带前缀的新文件。

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