
本文将提供一个清晰、简洁的Python解决方案,用于批量处理Excel文件,并在每个工作表(Sheet)中添加基于文件名的新列。通过使用pandas库和openpyxl引擎,可以高效地遍历Excel文件的所有工作表,并添加所需的信息,从而简化数据处理流程。
使用 Pandas 和 Openpyxl 为 Excel 文件所有 Sheet 添加列名
在数据处理过程中,经常需要对Excel文件进行批量操作,例如,在每个工作表中添加一列包含文件名信息的列。本文将详细介绍如何使用Python的pandas库和openpyxl引擎来实现这一目标。
步骤 1:导入必要的库
首先,需要导入os和pandas库。os库用于文件路径操作,pandas库用于Excel文件的读取和写入。
立即学习“Python免费学习笔记(深入)”;
import osimport pandas as pd
步骤 2:指定文件路径
设置包含Excel文件的根目录路径。请将 ‘XXXX’ 替换为实际的文件路径。
path = 'XXXX'
步骤 3:遍历文件
使用 os.walk() 函数遍历指定目录下的所有文件。然后,筛选出以 .xlsx 结尾的Excel文件。
for roots, dirs, files in os.walk(path): xlsfile = [f for f in files if f.endswith('.xlsx')]
步骤 4:处理每个 Excel 文件
对于找到的每个Excel文件,执行以下操作:
构建完整的文件路径。使用 pd.ExcelFile() 打开Excel文件。使用 pd.ExcelWriter() 创建一个写入器,指定使用 openpyxl 引擎。openpyxl 是一个用于读取/写入Excel 2010 xlsx/xlsm/xltx/xltm文件的Python库。遍历Excel文件的所有工作表(Sheet)。读取当前工作表的数据到 DataFrame。创建一个名为 filename 的新列,并根据文件名拆分的结果赋值。将修改后的 DataFrame 写回到当前工作表。
for xlsf in xlsfile: print(xlsf) excel_path = os.path.join(roots, xlsf) xls = pd.ExcelFile(excel_path) with pd.ExcelWriter(excel_path, engine='openpyxl') as writer: for sheet_name in xls.sheet_names: df = pd.read_excel(xls, sheet_name) df['filename'] = xlsf.split("-")[-2] df.to_excel(writer, sheet_name=sheet_name, index=False)
完整代码示例:
import osimport pandas as pdpath = 'XXXX'for roots, dirs, files in os.walk(path): xlsfile = [f for f in files if f.endswith('.xlsx')] for xlsf in xlsfile: print(xlsf) excel_path = os.path.join(roots, xlsf) xls = pd.ExcelFile(excel_path) with pd.ExcelWriter(excel_path, engine='openpyxl') as writer: for sheet_name in xls.sheet_names: df = pd.read_excel(xls, sheet_name) df['filename'] = xlsf.split("-")[-2] df.to_excel(writer, sheet_name=sheet_name, index=False)
注意事项:
请确保已安装 pandas 和 openpyxl 库。可以使用 pip install pandas openpyxl 命令进行安装。代码中的 xlsf.split(“-“)[-2] 用于从文件名中提取所需的信息。请根据实际的文件名格式进行调整。pd.ExcelWriter(excel_path, engine=’openpyxl’) 使用了 openpyxl 引擎,这对于处理较新的Excel文件格式(.xlsx)非常重要。index=False 参数用于防止将DataFrame的索引写入Excel文件。建议在处理大量文件之前,先用少量文件进行测试,以确保代码的正确性。
总结:
通过使用 pandas 库和 openpyxl 引擎,可以方便地批量处理Excel文件,并在每个工作表中添加自定义信息。这种方法不仅高效,而且可以灵活地适应不同的文件名格式和数据处理需求。在实际应用中,可以根据具体情况修改代码,以满足特定的业务需求。
以上就是在Python中为Excel文件的每个Sheet添加列名的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1374910.html
微信扫一扫
支付宝扫一扫