在Python中为Excel文件的每个Sheet添加列名

在python中为excel文件的每个sheet添加列名

本文将提供一个清晰、简洁的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

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2025年12月14日 14:33:23
下一篇 2025年12月14日 14:33:28

相关推荐

发表回复

登录后才能评论
关注微信