Python xlsxwriter库详解

1. 简介

Python是一种功能强大的编程语言，有许多优秀的第三方库可以帮助我们完成各种任务。其中，xlsxwriter是一个用于创建Excel电子表格的库，它通过编写XML文件，实现了在电子表格中生成各种元素（如单元格、图表等）的功能。

本文将详细介绍xlsxwriter库的安装与使用方法，并通过几个示例演示其基本功能。

2. 安装

使用pip命令可以轻松安装xlsxwriter库：

pip install XlsxWriter

3. 创建电子表格

3.1 创建Excel文件

我们可以使用xlsxwriter库创建一个新的Excel文件，代码示例如下：

import xlsxwriter

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

3.2 添加工作表

在新创建的Excel文件中，我们可以添加一个或多个工作表。默认情况下，一个工作表被创建在Excel文件中。下面是添加工作表的示例代码：

import xlsxwriter

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

# 添加一个工作表
worksheet = workbook.add_worksheet()

# 关闭Excel文件
workbook.close()

3.3 设置单元格格式

可以使用xlsxwriter库设置单元格的格式，包括字体、背景颜色、边框等。示例代码如下：

import xlsxwriter

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

# 创建带有格式的单元格
format = workbook.add_format({'bold': True, 'font_color': 'red'})

# 在单元格A1添加内容‘Hello World’并应用格式
worksheet.write('A1', 'Hello World', format)

# 关闭Excel文件
workbook.close()

3.4 写入数据

在创建的工作表中，我们可以写入各种数据类型，包括数字、字符串、日期等。示例代码如下：

import xlsxwriter

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

# 写入数据
worksheet.write(0, 0, 'Hello')
worksheet.write(0, 1, 'World')
worksheet.write(1, 0, 123)
worksheet.write(1, 1, 3.14)

# 关闭Excel文件
workbook.close()

3.5 设置行列宽度

可以使用xlsxwriter库设置工作表中的行列宽度。示例代码如下：

import xlsxwriter

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

# 设置第一行的高度为30
worksheet.set_row(0, 30)

# 设置第一列的宽度为20
worksheet.set_column(0, 0, 20)

# 关闭Excel文件
workbook.close()

3.6 添加图表

xlsxwriter库还支持在工作表中添加各种图表，如折线图、柱形图等。示例代码如下：

import xlsxwriter

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

# 写入数据
data = [1, 2, 3, 4, 5]
worksheet.write_column(0, 0, data)

# 添加折线图
chart = workbook.add_chart({'type': 'line'})
chart.add_series({'values': '=Sheet1! $A$ 1: $A$ 5'})
worksheet.insert_chart('C1', chart)

# 关闭Excel文件
workbook.close()

4. 示例演示

4.1 写入学生成绩

下面是一个示例，演示如何使用xlsxwriter库创建一个包含学生成绩的Excel电子表格：

import xlsxwriter

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

# 写入表头
worksheet.write(0, 0, '学号')
worksheet.write(0, 1, '姓名')
worksheet.write(0, 2, '语文成绩')
worksheet.write(0, 3, '数学成绩')

# 写入学生成绩
data = [
    ['001', '张三', 90, 85],
    ['002', '李四', 80, 92],
    ['003', '王五', 95, 88]
]

row = 1
col = 0

for student in data:
    worksheet.write(row, col, student[0])
    worksheet.write(row, col+1, student[1])
    worksheet.write(row, col+2, student[2])
    worksheet.write(row, col+3, student[3])
    row += 1

# 关闭Excel文件
workbook.close()

运行上述代码后，将生成一个名为example.xlsx的Excel电子表格文件，其中包含了学生成绩信息。

4.2 添加图表

下面是一个示例，演示如何使用xlsxwriter库在Excel工作表中添加柱状图：

import xlsxwriter

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

# 写入数据
worksheet.write(0, 0, '学科')
worksheet.write(0, 1, '分数')
worksheet.write(1, 0, '语文')
worksheet.write(1, 1, 90)
worksheet.write(2, 0, '数学')
worksheet.write(2, 1, 85)
worksheet.write(3, 0, '英语')
worksheet.write(3, 1, 95)

# 添加柱状图
chart = workbook.add_chart({'type': 'column'})
chart.add_series({'values': '=Sheet1! $B$ 2: $B$ 4'})
worksheet.insert_chart('D2', chart)

# 关闭Excel文件
workbook.close()

运行上述代码后，将生成一个名为example.xlsx的Excel电子表格文件，其中包含了一个柱状图，显示了不同学科的分数情况。