使用 XlsxWriter 模块 – Python

XlsxWriter 是一个用于在 Python 中写入 Excel 2007+ XLSX 文件的类库。它可以处理大量数据和格式，并支持图表、图片、条件格式和许多其他 Excel 功能。

安装

你可以使用 pip 安装 XlsxWriter 模块：

$ pip install XlsxWriter

创建 Excel 文档

要创建一个简单的 Excel 文档，你需要创建一个 Workbook 对象。Workbook 对象是一个 Excel 文件中的顶层对象，它包含一个或多个 Worksheet 对象。

import xlsxwriter

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

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

# 写入数据
worksheet.write('A1', 'Hello')
worksheet.write('B1', 'World')

# 关闭 Excel 文件
workbook.close()

这个程序创建了一个名为 example.xlsx 的 Excel 文件，并在工作表中写入了字符串 'Hello' 和 'World'。

创建格式

你可以使用 XlsxWriter 来自定义 Excel 中的单元格格式，例如字体、颜色、对齐和边框等。

import xlsxwriter

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

# 创建一个带有格式的工作表
bold = workbook.add_format({'bold': True})

worksheet = workbook.add_worksheet()
worksheet.write('A1', 'Hello', bold)
worksheet.write('B1', 'World')

# 关闭 Excel 文件
workbook.close()

在这个例子中，我们创建了一个带有粗体字的工作表，并在单元格 'A1' 中写入了字符串 'Hello'。

写入数据

要将数据写入 Excel 文件，你可以使用 write() 方法。可以写入数字、日期、字符串和公式。

import xlsxwriter

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

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

# 写入数字
worksheet.write(0, 0, 42)

# 写入日期
worksheet.write(1, 0, '2022-08-08')

# 写入公式
worksheet.write_formula(2, 0, '=SUM(1+2)')

# 关闭 Excel 文件
workbook.close()

在这个例子中，我们在单元格 'A1'、'A2' 和 'A3' 中写入了数字、字符串和公式。

增加图表

除了数据，你可以在 Excel 文件中添加图表以更好地可视化数据。XlsxWriter 支持很多种图表类型，包括条形图、饼图和散点图等。

import xlsxwriter

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

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

# 写入数据
data = [10, 20, 30, 40, 50]
worksheet.write_column('A1', data)

# 创建一个图表
chart = workbook.add_chart({'type': 'column'})

# 将数据添加到图表
chart.add_series({'values': '=Sheet1!$A$1:$A$5'})

# 将图表插入工作表
worksheet.insert_chart('C1', chart)

# 关闭 Excel 文件
workbook.close()

在这个例子中，我们创建了一个柱状图，并在工作表 'Sheet1' 中添加了数据。然后将数据添加到图表中，并将图表插入到工作表中。

结语

XlsxWriter 是一个功能强大的类库，可以用于创建和处理 Excel 文件。本文中我们介绍了如何创建 Excel 文档，如何创建格式化单元格、如何写入数据和如何添加图表。使用这些技巧，你可以轻松地创建复杂的 Excel 文件，以可视化数据。