Python如何自动获取目录下CSV文件名并作为变量传递

可以借助glob模块快速匹配目录下的CSV文件，结合你的需求，修改后的代码如下：

import pandas as pd
import glob

# 获取当前目录下所有CSV文件
csv_files = glob.glob("*.csv")

# 验证是否存在唯一CSV文件
if len(csv_files) != 1:
    raise ValueError("目录下需存在且仅存在一个CSV文件")

# 获取目标CSV文件名
target_csv = csv_files[0]

# 读取CSV文件
df = pd.read_csv(target_csv)
correct_df = df.copy()
correct_df.rename(columns={'Text1': 'Address1', 'Text2': 'Address2'}, inplace=True)

# 导出处理后的文件（避免覆盖原文件，可自定义命名规则）
output_filename = f"processed_{target_csv}"
correct_df.to_csv(output_filename, index=False, header=True)

关键说明： #

• glob.glob("*.csv")会匹配当前目录下所有后缀为.csv的文件，返回文件名列表
• 增加了判断逻辑，确保目录下只有一个CSV文件，避免出现意外情况
• 导出文件时用processed_前缀命名，防止覆盖原始CSV文件，你也可以根据需求修改导出文件名规则

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