Python openpyxl 获取指定表头所在列

Python openpyxl 获取指定表头所在列

在处理Excel文件的时候，有时我们需要根据表头的名称来获取对应的列索引，以便进一步处理数据。使用 openpyxl 这个 Python 库，我们可以方便地实现这个功能。在本文中，我将详细介绍如何利用 openpyxl 获取指定表头所在列的索引。

准备工作

首先，我们需要安装 openpyxl 这个库。你可以使用 pip 来安装：

pip install openpyxl

接下来，我们需要一个示例 Excel 文件。在本文中，我将使用一个名为 sample.xlsx 的文件作为示例。这个文件包含了一个名为 Sheet1 的工作表，其中有一些数据和表头。我们将使用 openpyxl 来打开这个文件并进行操作。

获取指定表头所在列的索引

下面是具体的步骤，来获取指定表头所在列的索引：

1. 导入 openpyxl 库:

from openpyxl import load_workbook

2. 打开 Excel 文件:

workbook = load_workbook("sample.xlsx")
sheet = workbook["Sheet1"]

3. 获取表头所在行的数据:

header_row = sheet[1]

4. 遍历表头数据，找到指定的表头名称，并获取其索引:

header_name = "Title"  # 指定要查找的表头名称
header_index = None

for idx, cell in enumerate(header_row, start=1):
    if cell.value == header_name:
        header_index = idx
        break

if header_index is not None:
    print(f"表头 {header_name} 所在列的索引为: {header_index}")
else:
    print(f"未找到表头 {header_name}")

在上面的代码中，我们首先打开了名为 sample.xlsx 的 Excel 文件，并选择了其中的 Sheet1 工作表。然后，我们获取了表头所在行的数据，并遍历这一行，找到了指定的表头名称所在的列索引。最后，我们打印出了找到的索引值。

运行结果

假设 sample.xlsx 文件中的数据如下所示：

如果我们运行上面的代码，并指定要查找的表头为 Title，则会输出：

表头 Title 所在列的索引为: 3

这表示 Title 表头所在的列索引为 3。

总结

在本文中，我们介绍了如何利用 openpyxl 这个 Python 库来获取指定表头所在列的索引。通过简单的几步操作，我们可以轻松地实现这个功能，从而更加方便地处理 Excel 文件中的数据。