准备工作
在开始之前,确保你的 Python 环境中已经安装了 openpyxl
和 pandas
库。可以使用以下命令进行安装:
pip install openpyxl pandas
第一步:编写函数保留表头和第一行数据
我们首先编写一个函数 keep_first_two_rows
,用于保留指定工作表的表头和第一行数据,其余数据将被删除。
from openpyxl import load_workbook def keep_first_two_rows(filepath, sheetname): try: # 加载 Excel 文件 wb = load_workbook(filepath) ws = wb[sheetname] # 删除第三行之后的所有行 ws.delete_rows(3, ws.max_row) # 保存修改后的 Excel 文件 wb.save(filepath) print(f"成功保留表头和第一行数据到工作表 {sheetname}") except Exception as e: print(f"保留表头和第一行数据时出错: {e}")
解释:
load_workbook(filepath)
:加载指定路径的 Excel 文件。wb[sheetname]
:获取指定名称的工作表。ws.delete_rows(3, ws.max_row)
:从第三行开始删除所有后续行,只保留前两行。