当前位置：首页 > article >正文

python在excel表某一列之后插入六列并命名名字

article 2025/3/1 19:12:40

要在 Python 中使用 openpyxl 包在 Excel 表的某一列之后插入六列并命名这些新列，可以按照以下步骤进行操作。这包括加载 Excel 文件、插入新列，并为新列添加标题名称。

安装 `openpyxl`

确保已经安装 openpyxl 如果还没有安装，可以使用以下命令进行安装：

pip install openpyxl

示例代码

以下代码示例展示了如何在某一列（例如 B 列）之后插入六列，并为每个新列添加标题。

1. 原始数据 (`example.xlsx`) 示例

A	B	C	D
1	2	3	4
5	6	7	8

2. 实现步骤和代码

修改后的数据目标

A	B	NewCol1	NewCol2	NewCol3	NewCol4	NewCol5	NewCol6	C	D
1	2							3	4
5	6							7	8

实现过程

from openpyxl import load_workbook

def insert_columns_after(sheet, col_index, num_cols, col_names):
    """
    在指定列之后插入几列，并命名新列
    :param sheet: 工作表对象
    :param col_index: 指定列的列号（从1开始）
    :param num_cols: 插入的列数
    :param col_names: 新列的标题列表
    """
    if len(col_names) != num_cols:
        raise ValueError("Number of column names must match number of columns to insert")
    
    # 插入新列
    sheet.insert_cols(col_index + 1, num_cols)
    
    # 获取第一行
    row = sheet[1]
    
    # 设置新列的标题
    for i in range(num_cols):
        row[col_index + 1 + i].value = col_names[i]

# 加载工作簿和工作表
wb = load_workbook('example.xlsx')
ws = wb.active  # 假设我们操作的是活动表，如果需要操作特定表，请使用 ws = wb['SheetName']

# 在第 2 列（B 列）之后插入 6 列，并设置新列的名称
new_column_names = [f"NewCol{i+1}" for i in range(6)]
insert_columns_after(ws, 2, 6, new_column_names)

# 保存修改后的工作簿
wb.save('example_modified.xlsx')

详细说明

加载工作簿和工作表：

使用 load_workbook 函数加载 Excel 文件，并选择要操作的工作表（这里使用的是活动表）。
```
wb = load_workbook('example.xlsx')
ws = wb.active
```
定义插入列的函数：

定义一个 insert_columns_after 函数，用于在指定列之后插入多列，并设置新列的名称。
```
def insert_columns_after(sheet, col_index, num_cols, col_names):
    if len(col_names) != num_cols:
        raise ValueError("Number of column names must match number of columns to insert")
    
    # 插入新列
    sheet.insert_cols(col_index + 1, num_cols)
    
    # 获取第一行
    row = sheet[1]
    
    # 设置新列的标题
    for i in range(num_cols):
        row[col_index + 1 + i].value = col_names[i]
```
- sheet：工作表对象。
- col_index：指定的列号，从 1 开始。
- num_cols：需要插入的列数。
- col_names：新列的标题列表，长度必须与 num_cols 相等。
- sheet.insert_cols(col_index + 1, num_cols)：使用 insert_cols 方法在指定列后插入新列。
- row = sheet[1]：获取工作表的第一行。
- row[col_index + 1 + i].value = col_names[i]：设置新插入列的标题。
调用函数插入列并命名：

在 B 列之后插入六列，同时设置新列的名称。这里 col_index 参数传递为 2（B 列），num_cols 参数传递为 6，col_names 自定义为 NewCol1 到 NewCol6。
```
new_column_names = [f"NewCol{i+1}" for i in range(6)]
insert_columns_after(ws, 2, 6, new_column_names)
```
保存修改后的工作簿：

将修改后的工作簿保存为一个新文件，避免覆盖原文件。
```
wb.save('example_modified.xlsx')
```