使用Python读取和修改Excel文件(基于xlrd、xlwt和openpyxl模块)

下面详细讲解如何使用Python读取和修改Excel文件。

1. 介绍

Excel是一种广泛使用的电子表格软件，而Python是一种流行的编程语言。Python中有许多可以帮助我们读取和修改Excel文件的库。本教程将重点介绍三个最受欢迎的库：xlrd、xlwt和openpyxl。

• xlrd：用于读取Excel文件，支持.xls和.xlsx格式。
• xlwt：用于创建和修改Excel文件，支持.xls格式。
• openpyxl：用于创建和修改Excel文件，支持.xlsx格式。

请注意，xlwt只能处理旧版本的.xls文件，openpyxl只能处理新版本的.xlsx文件，而xlrd可以处理旧版本和新版本的Excel文件。

2. 安装

在使用这些库之前，我们需要先安装它们。可以使用pip命令来安装它们，命令如下：

pip install xlrd xlwt openpyxl

3. 读取Excel文件

以下代码演示了如何使用xlrd读取Excel文件：

import xlrd

# 打开Excel文件
workbook = xlrd.open_workbook('example.xlsx')

# 获取所有工作表名
print(workbook.sheet_names())

# 获取第一个工作表
sheet = workbook.sheet_by_index(0)

# 获取行数和列数
nrows, ncols = sheet.nrows, sheet.ncols
print(f"Number of rows: {nrows}, Number of columns: {ncols}")

# 遍历打印每一行数据
for i in range(nrows):
    row = sheet.row_values(i)
    print(row)

以上代码打印了Excel文件的工作表名、行数和列数，以及每一行的数据。你可以根据需要修改代码来提取所需的数据。

4. 创建和修改Excel文件

以下代码演示了如何使用openpyxl创建和修改Excel文件：

import openpyxl

# 创建一个新的工作簿
workbook = openpyxl.Workbook()

# 获取第一个工作表
sheet = workbook.active

# 填写标题行
sheet['A1'] = '姓名'
sheet['B1'] = '年龄'
sheet['C1'] = '性别'

# 填写数据行
sheet.append(['张三', 20, '男'])
sheet.append(['李四', 25, '女'])
sheet.append(['王五', 30, '男'])

# 保存工作簿
workbook.save('example.xlsx')

以上代码创建了一个新的.xlsx文件，并在其中创建第一个工作表，然后填写标题行和数据行，并将其保存为Excel文件。你可以根据需要修改代码来创建自己的Excel文件。

以下代码演示了如何使用openpyxl修改Excel文件：

import openpyxl

# 打开Excel文件
workbook = openpyxl.load_workbook('example.xlsx')

# 获取第一个工作表
sheet = workbook.active

# 获取行数和列数
nrows, ncols = sheet.max_row, sheet.max_column
print(f"Number of rows: {nrows}, Number of columns: {ncols}")

# 插入新行
new_row = ['赵六', 35, '女']
sheet.append(new_row)

# 修改数据
sheet['B2'] = 26

# 保存工作簿
workbook.save('example.xlsx')

以上代码打开了一个.xlsx文件，并修改了其中的数据，然后保存文件。你也可以根据需要修改代码来修改自己的Excel文件。

5. 示例说明

以下是两个简单的示例说明：

示例1：计算Excel文件中某一列的总和

假设我们有一个.xlsx文件，其中包含一个名为Sheet1的工作表，其中第一列是数值。我们想计算该列的总和。以下是实现这个目标的代码：

import openpyxl

# 打开Excel文件
workbook = openpyxl.load_workbook('example.xlsx')

# 获取第一个工作表
sheet = workbook['Sheet1']

# 计算列的总和
total = 0
for cell in sheet['A']:
    if isinstance(cell.value, (int, float)):
        total += cell.value

print(f"The sum of column A is {total}")

以上代码计算了列A的总和，并打印结果。

示例2：将Excel文件中的特定数据导出为CSV文件

假设我们有一个.xlsx文件，其中包含一个名为Sheet1的工作表，我们想将其中某一列特定值的数据导出为CSV文件。以下是实现这个目标的代码：

import csv
import openpyxl

# 打开Excel文件
workbook = openpyxl.load_workbook('example.xlsx')

# 获取第一个工作表
sheet = workbook['Sheet1']

# 导出特定值的数据到CSV文件
with open('output.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(['ID', 'Name'])  # CSV文件的标题行
    for row in sheet.iter_rows(values_only=True):
        if row[2] == 'male':  # 只保留性别为男性的记录
            writer.writerow([row[0], row[1]])  # 写入CSV文件中

以上代码读取.xlsx文件中的工作表数据，并将性别为男性的记录导出到CSV文件中。你可以根据需要修改代码来导出自己文件中的数据。

6. 总结

本文介绍了如何使用xlrd、xlwt和openpyxl三个库来读取和修改Excel文件。我们演示了如何使用这些库来读取和创建Excel文件，并提供了两个示例说明。如果你需要在Python中处理Excel文件，请尝试使用这些库，它们将为你节省大量时间。