代码收藏家技术教程 2025-01-11

Python openpyxl 如何删除行和列

在数据处理和分析中，Excel 文件是不可或缺的一部分。Python 的 openpyxl 库提供了强大的功能来操作 Excel 文件，包括读取、写入、修改和格式化数据。然而，很多初学者在使用 openpyxl 时会遇到一个常见的问题：如何删除 Excel 文件中的行和列？本文将详细介绍 openpyxl 中删除行和列的方法，并通过具体的代码示例帮助你轻松掌握这一技能。

为什么需要删除行和列？

在实际工作中，我们经常需要对 Excel 文件进行数据清洗。例如，去除多余的空行、删除不需要的列或者清理错误的数据。这些操作不仅能够提高数据的质量，还能提升后续分析的效率。因此，学会如何在 openpyxl 中删除行和列是非常重要的。

安装 openpyxl

如果你还没有安装 openpyxl，可以使用以下命令进行安装：

pip install openpyxl

删除行

基本方法

openpyxl 提供了 delete_rows 方法来删除指定的行。该方法的基本语法如下：

worksheet.delete_rows(idx, amount=None)

idx：要删除的行号。

amount：可选参数，表示要删除的行数。如果不指定，默认只删除一行。

示例 1：删除单行

假设我们有一个 Excel 文件 example.xlsx，其中包含以下数据：

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

我们想要删除第 2 行（即索引为 2 的行），可以使用以下代码：

from openpyxl import load_workbook

# 加载工作簿
wb = load_workbook('example.xlsx')
ws = wb.active

# 删除第 2 行
ws.delete_rows(2)

# 保存工作簿
wb.save('example_deleted_row.xlsx')

执行上述代码后，example_deleted_row.xlsx 文件的内容将变为：

A	B	C
1	2	3
7	8	9

示例 2：删除多行

如果需要删除多行，可以通过指定 amount 参数来实现。例如，删除从第 2 行开始的 2 行：

# 删除从第 2 行开始的 2 行
ws.delete_rows(2, 2)

# 保存工作簿
wb.save('example_deleted_multiple_rows.xlsx')

执行上述代码后，example_deleted_multiple_rows.xlsx 文件的内容将变为：

A	B	C
1	2	3

动态删除行

在实际应用中，我们可能需要根据某些条件动态删除行。例如，删除所有值为 0 的行。可以使用以下代码实现：

# 遍历所有行，删除值为 0 的行
for row in list(ws.iter_rows(min_row=1, max_row=ws.max_row)):
    if all(cell.value == 0 for cell in row):
        ws.delete_rows(row[0].row)

# 保存工作簿
wb.save('example_deleted_zero_rows.xlsx')

删除列

基本方法

openpyxl 提供了 delete_cols 方法来删除指定的列。该方法的基本语法如下：

worksheet.delete_cols(idx, amount=None)

idx：要删除的列号。

amount：可选参数，表示要删除的列数。如果不指定，默认只删除一列。

示例 1：删除单列

假设我们有一个 Excel 文件 example.xlsx，其中包含以下数据：

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

我们想要删除第 2 列（即索引为 2 的列），可以使用以下代码：

from openpyxl import load_workbook

# 加载工作簿
wb = load_workbook('example.xlsx')
ws = wb.active

# 删除第 2 列
ws.delete_cols(2)

# 保存工作簿
wb.save('example_deleted_column.xlsx')

执行上述代码后，example_deleted_column.xlsx 文件的内容将变为：

A	C
1	3
4	6
7	9

示例 2：删除多列

如果需要删除多列，可以通过指定 amount 参数来实现。例如，删除从第 2 列开始的 2 列：

# 删除从第 2 列开始的 2 列
ws.delete_cols(2, 2)

# 保存工作簿
wb.save('example_deleted_multiple_columns.xlsx')

执行上述代码后，example_deleted_multiple_columns.xlsx 文件的内容将变为：

A
1
4
7

动态删除列

在实际应用中，我们可能需要根据某些条件动态删除列。例如，删除所有值为 0 的列。可以使用以下代码实现：

# 遍历所有列，删除值为 0 的列
for col in reversed(list(ws.iter_cols(min_col=1, max_col=ws.max_column))):
    if all(cell.value == 0 for cell in col):
        ws.delete_cols(col[0].column)

# 保存工作簿
wb.save('example_deleted_zero_columns.xlsx')