Python读取Excel数据某一列|极客教程

Python读取Excel数据某一列

1. 引言

在实际工作中，我们经常需要对Excel中的数据进行处理和分析。而Python作为一种功能强大的编程语言，提供了许多处理Excel数据的库，如openpyxl、xlrd等。本文将详细介绍如何使用Python读取Excel中的某一列数据，并给出示例代码和运行结果。

2. 准备工作

在开始之前，我们需要安装openpyxl库。可以使用以下命令来安装：

pip install openpyxl

3. 读取Excel数据

首先，我们需要导入openpyxl库，并通过load_workbook方法加载Excel文件：

from openpyxl import load_workbook

# 加载Excel文件
wb = load_workbook('data.xlsx')

接下来，我们需要选择要读取数据的表格。一般来说，Excel文件中的每个表格都有一个名字，我们可以通过wb.sheetnames属性获取到所有表格的名字：

# 获取所有表格的名字
sheet_names = wb.sheetnames

# 选择第一个表格
sheet = wb[sheet_names[0]]

现在，我们已经成功选择了表格，接下来我们需要选择要读取的列。由于Excel中的列是通过字母来标识的（如A、B、C…），我们需要将字母转换成对应的数字。可以使用如下代码进行转换：

# 字母转数字：A -> 1, B -> 2, ...
def col2num(col_str):
    base = ord('A')
    col_num = 0
    for char in col_str:
        col_num = col_num * 26 + (ord(char) - base + 1)
    return col_num

# 数字转字母：1 -> A, 2 -> B, ...
def num2col(col_num):
    base = ord('A') - 1
    col_str = ''
    while col_num > 0:
        remainder = col_num % 26
        if remainder == 0:
            col_str = 'Z' + col_str
            col_num = col_num // 26 - 1
        else:
            col_str = chr(remainder + base) + col_str
            col_num = col_num // 26
    return col_str

假设我们要读取的列为第二列（即B列），我们可以通过以下代码读取该列的数据：

# 读取B列的数据
column = 'B'
column_num = col2num(column)

data = []
for row in sheet.iter_rows(values_only=True):
    data.append(row[column_num - 1])

这样，我们就成功读取了Excel中的某一列数据，并存储在data列表中。

4. 完整示例代码

下面是一个完整的示例代码，展示了如何使用Python读取Excel中的某一列数据：

from openpyxl import load_workbook

# 字母转数字：A -> 1, B -> 2, ...
def col2num(col_str):
    base = ord('A')
    col_num = 0
    for char in col_str:
        col_num = col_num * 26 + (ord(char) - base + 1)
    return col_num

# 数字转字母：1 -> A, 2 -> B, ...
def num2col(col_num):
    base = ord('A') - 1
    col_str = ''
    while col_num > 0:
        remainder = col_num % 26
        if remainder == 0:
            col_str = 'Z' + col_str
            col_num = col_num // 26 - 1
        else:
            col_str = chr(remainder + base) + col_str
            col_num = col_num // 26
    return col_str

# 加载Excel文件
wb = load_workbook('data.xlsx')

# 获取所有表格的名字
sheet_names = wb.sheetnames

# 选择第一个表格
sheet = wb[sheet_names[0]]

# 读取B列的数据
column = 'B'
column_num = col2num(column)

data = []
for row in sheet.iter_rows(values_only=True):
    data.append(row[column_num - 1])

# 打印数据
for value in data:
    print(value)