Python自动化办公对每个子文件夹的Excel表加个表头(Excel不同名且有xls文件)

Python进阶者

发布于 2023-09-02 09:47:33

1590

发布于 2023-09-02 09:47:33

文章被收录于专栏：Python爬虫与数据挖掘Python爬虫与数据挖掘

一、前言

上一篇文章，我们抛出了一个问题，这篇文章来进行解答。如果针对子文件夹下不同的Excel表名，而且Excel表格类型包括了.xls和.xlsx应该如何处理？要求一步到位。

二、实现过程

其实这里依靠【ChatGPT】给的答案，只需要修改其中一行代码就可以搞定了，如下所示：

import os
import pandas as pd

# 定义文件夹路径
folder_path = r"C:\Users\pdcfi\Desktop\新建文件夹"

# 获取所有子文件夹路径
subfolders = [f.path for f in os.scandir(folder_path) if f.is_dir()]

# 为每个Excel表格添加表头并保存
for subfolder in subfolders:
    # 获取该子文件夹中所有Excel表格的路径
    excel_paths = [f.path for f in os.scandir(subfolder) if f.is_file() and f.name.endswith(".xlsx") or f.name.endswith(".xls")]
    for excel_path in excel_paths:
        # 读取Excel表格
        df = pd.read_excel(excel_path, header=None)
        # 添加表头
        df.columns = ['经度', '纬度']
        # 保存Excel表格
        df.to_excel(excel_path, index=False)

在上述代码中，首先定义了文件夹路径。然后使用os.scandir()函数获取该文件夹下的所有子文件夹路径。接下来，遍历每个子文件夹，使用os.scandir()函数获取该子文件夹中所有Excel表格的路径，然后使用pandas模块的read_excel()函数读取Excel表格。接着，为Excel表格添加表头，使用to_excel()函数保存Excel表格。最终，每个Excel表格都会被添加表头并保存。

希望这个代码可以满足您的需求！

这样的话，就直接一步到位了，顺利地解决了粉丝的问题。