如何使CSV输出文件保留原文件名并添加自定义后缀?
解决CSV文件输出保留原文件名并添加自定义后缀的问题
直接修改循环内的文件名处理逻辑,提取原文件名后拼接自定义后缀即可,以下是修改后的完整代码:
import pandas as pd import numpy as np import glob import os all_files = glob.glob("C:/Users/Gamer/Documents/Colbun/Saturn/*.csv") for f in all_files: df = pd.read_csv(f, header=0, usecols=["t","f"]) # 提取原文件名(不含路径和扩展名) raw_filename = os.path.splitext(os.path.basename(f))[0] # 构造带自定义后缀的输出文件名 output_filename = f"{raw_filename}_ext.csv" # 拼接输出文件夹路径与文件名 output_path = os.path.join("C:/Users/Gamer/Documents/Colbun/Saturn2", output_filename) # 保存文件,可选index=False避免生成多余的索引列 df.to_csv(output_path, index=False)
关键逻辑说明
os.path.basename(f):从完整文件路径中提取带扩展名的文件名(例:C:/.../aaa.csv→aaa.csv)os.path.splitext(...):将文件名拆分为"主文件名"和"扩展名"两部分(例:aaa.csv→('aaa', '.csv')),取第一部分得到无后缀的原文件名os.path.join:规范拼接路径与文件名,避免手动拼接字符串带来的路径分隔符问题,适配Windows/macOS/Linux不同系统index=False:可选参数,若不需要将DataFrame的索引保存为CSV的一列,建议添加该参数
内容的提问来源于stack exchange,提问作者Isaac Rojas Oyarzun
相关产品推荐
相关产品推荐

