Python是处理Excel文件的一种非常强大且灵活的工具,尤其是通过使用openpyxl
和xlrd
等库。openpyxl
专注于Excel 2007及更高版本的.xlsx
文件的创建、读取、修改和写入,而xlrd
则主要用于读取早期版本的Excel文件(.xls
),但自从2020年xlrd
库不再支持.xlsx
文件,因此它的主要用途现在局限于读取.xls
文件。
一、安装库
在使用这些库之前,我们需要先安装它们。可以通过以下命令安装:
pip install openpyxl
pip install xlrd
openpyxl
是一个更现代的库,主要用于处理.xlsx
格式的文件,而xlrd
在新版更新后仅支持.xls
文件的读取操作。
二、openpyxl
库的使用
1. 创建Excel文件
首先,我们可以通过openpyxl
创建一个新的Excel文件,并向其中写入数据:
from openpyxl import Workbook
# 创建一个新的工作簿
wb = Workbook()
# 激活默认的工作表
ws = wb.active
# 写入数据到工作表
ws['A1'] = 'Hello'
ws['B1'] = 'World'
# 保存工作簿
wb.save("example.xlsx")
在上面的代码中,我们首先创建了一个工作簿,然后激活了默认的工作表,并向A1和B1单元格中写入了数据,最后将工作簿保存为example.xlsx
。
2. 读取Excel文件
openpyxl
也可以用于读取现有的Excel文件:
from openpyxl import load_workbook
# 加载已有的工作簿
wb = load_workbook("example.xlsx")
# 选择活动工作表
ws = wb.active
# 读取单元格数据
print(ws['A1'].value) # 输出:Hello
print(ws['B1'].value) # 输出:World
load_workbook()
函数用于加载现有的工作簿,并返回一个Workbook
对象,然后可以通过.active
属性访问默认的工作表。
3. 操作工作表
在openpyxl
中,我们可以轻松地操作工作表,例如创建新的工作表、修改工作表名称、删除工作表等:
# 创建新工作表
ws1 = wb.create_sheet("NewSheet")
# 修改工作表名称
ws1.title = "MySheet"
# 删除工作表
wb.remove(ws1)
# 或者使用 del 关键字
del wb["MySheet"]
4. 写入和修改数据
openpyxl
允许我们使用不同的方式写入数据,除了通过单元格坐标,还可以通过遍历行和列的方式批量操作数据:
# 批量写入数据
for row in range(1, 10):
for col in range(1, 5):
ws.cell(row=row, column=col, value=f"Row {row}, Col {col}")
# 修改单元格的值
ws['A1'] = 'Updated Value'
在上面的代码中,我们通过ws.cell()
方法可以指定行和列,使用嵌套循环快速填充数据。
5. 单元格样式
除了读写数据外,openpyxl
还支持修改单元格的样式,例如字体、边框、颜色等:
from openpyxl.styles import Font, Color, Alignment
# 设置单元格字体
ws['A1'].font = Font(size=14, bold=True)
# 设置单元格对齐方式
ws['A1'].alignment = Alignment(horizontal='center', vertical='center')
# 设置单元格填充颜色
from openpyxl.styles import PatternFill
ws['A1'].fill = PatternFill("solid", fgColor="00FF00")
openpyxl.styles
模块提供了丰富的样式选项,允许我们根据需求定制Excel文件的外观。
6. 处理公式
openpyxl
还支持Excel中的公式。例如:
# 添加一个简单的加法公式
ws['C1'] = '=A1+B1'
# 手动计算公式(注意:需要 Excel 应用程序打开此文件后才能真正计算)
wb.save("formula_example.xlsx")
虽然openpyxl
可以写入和读取公式,但实际的计算需要Excel应用程序本身来执行。
三、xlrd
库的使用
xlrd
专注于读取Excel文件,特别是xls
格式。虽然它不再支持读取xlsx
文件,但对于处理旧版的Excel文件依然很有用。
1. 读取Excel文件
使用xlrd
读取Excel文件非常简单:
import xlrd
# 打开一个Excel文件
book = xlrd.open_workbook('example.xls')
# 获取第一个工作表
sheet = book.sheet_by_index(0)
# 读取单元格数据
print(sheet.cell_value(0, 0)) # 输出第1行第1列的数据
这里我们通过xlrd.open_workbook()
函数打开了一个.xls
文件,然后通过sheet_by_index()
方法获取工作表,并通过sheet.cell_value()
读取单元格的内容。
2. 获取工作表信息
xlrd
可以帮助我们获取更多关于工作表的信息,例如行数、列数等:
# 获取工作表名称
print(sheet.name)
# 获取工作表的行数和列数
print(sheet.nrows)
print(sheet.ncols)
这对于遍历整个表格的内容是非常有用的。
3. 遍历单元格数据
xlrd
允许我们轻松遍历整个工作表中的数据:
# 遍历行数据
for row_idx in range(sheet.nrows):
row = sheet.row(row_idx)
for col_idx, cell in enumerate(row):
print(f"({row_idx}, {col_idx}) = {cell.value}")
上面的代码示例展示了如何遍历工作表中的所有数据,逐行输出每个单元格的内容。
四、openpyxl
与xlrd
的配合使用
在某些场景下,我们可能需要处理.xls
和.xlsx
两种文件格式。这时,可以结合使用openpyxl
和xlrd
。例如,我们可以使用xlrd
读取.xls
文件,然后使用openpyxl
将其转换为.xlsx
格式:
import xlrd
from openpyxl import Workbook
# 使用xlrd打开xls文件
xls_book = xlrd.open_workbook('example.xls')
xls_sheet = xls_book.sheet_by_index(0)
# 创建一个新的xlsx文件
wb = Workbook()
ws = wb.active
# 将xls数据复制到xlsx
for row_idx in range(xls_sheet.nrows):
for col_idx in range(xls_sheet.ncols):
ws.cell(row=row_idx+1, column=col_idx+1, value=xls_sheet.cell_value(row_idx, col_idx))
# 保存新的xlsx文件
wb.save('converted_example.xlsx')
这样,通过结合xlrd
和openpyxl
,我们可以轻松实现不同Excel格式之间的转换和处理。
openpyxl
主要用于处理现代的.xlsx
文件,而xlrd
则用于读取老旧的.xls
文件格式。通过结合这两个库,我们可以灵活地处理各种Excel文件格式,并根据需要进行数据转换和格式定制。
在实际应用中,根据文件格式的不同选择合适的库进行操作,可以提高代码的效率和兼容性。掌握这两个库的使用,将为数据处理和分析领域提供极大的便利。