Python openpyxl 获取指定表头所在列

Python openpyxl 获取指定表头所在列

在处理Excel表格数据时，经常会遇到需要根据表头来获取特定列数据的情况。Python中常用的库openpyxl可以帮助我们实现这一功能。本文将介绍如何使用openpyxl库来获取指定表头所在列的数据。

1. 安装openpyxl库

如果你还没有安装openpyxl库，可以使用pip命令来安装：

pip install openpyxl

2. 加载Excel文件

首先，我们需要加载Excel文件并打开指定的工作表。假设我们有一个名为data.xlsx的Excel文件，其中包含一个名为Sheet1的工作表，代码如下：

from openpyxl import load_workbook

# 加载Excel文件
wb = load_workbook('data.xlsx')
# 打开Sheet1工作表
ws = wb['Sheet1']

3. 获取表头所在行

接下来，我们需要找到包含指定表头的行。假设我们要获取表头为Name的列数据，代码如下：

header_row = 1  # 假设表头在第1行
header_column = None

for row in ws.iter_rows(min_row=header_row, max_row=header_row):
    for cell in row:
        if cell.value == 'Name':
            header_column = cell.column
            break

if header_column is not None:
    print(f"表头'Name'在第{header_column}列")
else:
    print("未找到指定的表头")

在上面的代码中，我们先遍历第1行的单元格，找到值为Name的单元格的列号，然后获取该列号，并打印出表头Name在第几列。

4. 获取指定列数据

有了表头所在列的信息后，我们就可以通过该列号来获取整列数据。代码如下：

column_data = []

for row in ws.iter_rows(min_row=header_row + 1):
    cell = row[header_column - 1]
    column_data.append(cell.value)

print(f"指定列数据为：{column_data}")

在上面的代码中，我们从第header_row + 1行开始遍历，获取指定列的数据，并存储在column_data列表中。最后打印出该列的所有数据。

5. 完整代码示例

下面是整合以上代码的完整示例：

from openpyxl import load_workbook

# 加载Excel文件
wb = load_workbook('data.xlsx')
# 打开Sheet1工作表
ws = wb['Sheet1']

# 获取表头所在行
header_row = 1  # 假设表头在第1行
header_column = None

for row in ws.iter_rows(min_row=header_row, max_row=header_row):
    for cell in row:
        if cell.value == 'Name':
            header_column = cell.column
            break

if header_column is not None:
    print(f"表头'Name'在第{header_column}列")

    # 获取指定列数据
    column_data = []

    for row in ws.iter_rows(min_row=header_row + 1):
        cell = row[header_column - 1]
        column_data.append(cell.value)

    print(f"指定列数据为：{column_data}")
else:
    print("未找到指定的表头")

6. 运行结果

假设我们的Excel文件data.xlsx中的数据如下：

运行以上代码后，输出为：

表头'Name'在第2列
指定列数据为：['Alice', 'Bob', 'Carol', 'David']

通过以上代码示例，我们成功地使用openpyxl库获取了指定表头所在列的数据。在实际应用中，可以根据需要修改表头名称和Excel文件名，灵活地应用到不同的场景中。

本文链接：http://so.lmcjl.com/news/4793/