Python 处理 Excel 文件之 XlsxWriter 模块

简介

XlsxWriter 是一个使用纯 Python 编写的强大的 Excel 写入库。通过它，我们可以创建和修改 Excel 文档，支持多种自定义样式，如单元格格式、字体、颜色、边框等等。XlsxWriter 还支持创建图表、图表系列、数据有效性等。

安装

通过 pip 可以很容易地安装 XlsxWriter 模块。

pip install XlsxWriter

创建 Excel 文档

首先，我们需要导入 XlsxWriter 模块。

import xlsxwriter

之后，我们可以通过下面的代码创建并保存一个空白的 Excel 文档。

# 创建一个 Excel 文档
workbook = xlsxwriter.Workbook('example.xlsx')

# 创建一个工作表
worksheet = workbook.add_worksheet()

# 关闭并保存 Excel 文档
workbook.close()

我们可以分别使用 xlsxwriter.Workbook 和 workbook.add_worksheet 函数创建一个 Excel 文档和工作表。接下来，我们可以在工作表中创建单元格并写入数据。

# 创建一个 Excel 文档
workbook = xlsxwriter.Workbook('example.xlsx')

# 创建一个工作表
worksheet = workbook.add_worksheet()

# 创建一个带样式的单元格
bold = workbook.add_format({'bold': True})
worksheet.write('A1', 'Hello', bold)
worksheet.write('B1', 'world')

# 关闭并保存 Excel 文档
workbook.close()

可以看到，我们使用 workbook.add_format 函数定义了一个加粗的字体样式，并在 worksheet.write 函数中使用 bold 参数将单元格样式设置为加粗。

我们也可以使用循环、列表、字典等数据结构来动态地写入数据，如下所示：

# 创建一个 Excel 文档
workbook = xlsxwriter.Workbook('example.xlsx')

# 创建一个工作表
worksheet = workbook.add_worksheet()

# 写入一组数据
data = ['Foo', 'Bar', 'Baz']
row = 0
for col, cell_data in enumerate(data):
    worksheet.write(row, col, cell_data)

# 写入多组数据
datas = [
    ['Alice', 18],
    ['Bob', 20],
    ['Charlie', 22],
]
for row, row_data in enumerate(datas):
    for col, cell_data in enumerate(row_data):
        worksheet.write(row, col, cell_data)

# 关闭并保存 Excel 文档
workbook.close()

在上述代码中，我们使用循环和列表来动态地写入一组和多组数据。之后，我们可以在 Excel 文档中查看输出的数据。