Python实现Excel表格批量添加列名

2024-01-29 10:56:43

概述

在日常办公中，我们经常需要对大量Excel表格进行处理，其中一项常见任务就是为表格添加新列，并将该列的内容设置为表格名。传统的方法是手动逐个添加，这对于少量表格来说还好，但当表格数量较多时，就会变得非常繁琐和耗时。

为了解决这个问题，我们可以借助Python的强大功能，编写一个自动化脚本，批量实现在Excel表格中添加新列并填充表格名。这样，我们可以节省大量时间和精力，并将工作重点放在更有价值的任务上。

先决条件

在开始之前，您需要确保满足以下先决条件：

已安装Python 3或更高版本。
已安装openpyxl库。您可以使用以下命令安装openpyxl：

pip install openpyxl

拥有要处理的Excel表格，并确保它们都位于同一个文件夹中。

步骤

1. 导入必要的库

首先，我们需要导入必要的Python库，包括openpyxl和os库。openpyxl库用于操作Excel表格，而os库用于获取文件系统信息。

import openpyxl
import os

2. 获取Excel表格列表

接下来，我们需要获取要处理的Excel表格列表。我们可以使用os.listdir()函数获取当前目录下的所有文件，并过滤出那些以“.xlsx”或“.xls”结尾的文件。

# 获取当前目录下的所有Excel表格
excel_files = [f for f in os.listdir(".") if f.endswith(".xlsx") or f.endswith(".xls")]

3. 循环处理Excel表格

现在，我们可以循环处理每个Excel表格。对于每个表格，我们需要打开它，添加新列，并填充表格名。

for excel_file in excel_files:
    # 打开Excel表格
    wb = openpyxl.load_workbook(excel_file)

    # 获取第一个工作表
    sheet = wb.active

    # 添加新列
    sheet.insert_cols(sheet.max_column + 1)

    # 填充表格名
    sheet.cell(1, sheet.max_column).value = excel_file.replace(".xlsx", "").replace(".xls", "")

    # 保存工作簿
    wb.save(excel_file)