python 数据生成excel导出（xlwt,wlsxwrite）代码实例

下面我就来详细讲解一下 "Python数据生成Excel导出（xlwt，xlsxwriter）代码实例" 的完整实例教程。

1. 安装必要的库

在使用xlwt和xlsxwriter之前，需要先安装相应的库。可以通过pip命令来安装，例如：

pip install xlwt
pip install xlsxwriter

2. 创建Excel文件

使用xlwt创建Excel文件

使用xlwt库创建Excel文件非常简单。下面的代码展示了如何创建一个Excel文件，并在其中添加一个名为“Sheet1”的工作表，并向该工作表中添加一些数据。

import xlwt

# 创建Excel文件
workbook = xlwt.Workbook()

# 添加工作表
worksheet = workbook.add_sheet('Sheet1')

# 添加数据
worksheet.write(0, 0, '姓名')
worksheet.write(0, 1, '年龄')
worksheet.write(1, 0, '张三')
worksheet.write(1, 1, 18)

# 保存Excel文件
workbook.save('test.xls')

使用xlsxwriter创建Excel文件

使用xlsxwriter库创建Excel文件同样非常简单。下面的代码展示了如何创建一个Excel文件，并在其中添加一个名为“Sheet1”的工作表，并向该工作表中添加一些数据。

import xlsxwriter

# 创建Excel文件
workbook = xlsxwriter.Workbook('test.xlsx')

# 添加工作表
worksheet = workbook.add_worksheet('Sheet1')

# 添加数据
worksheet.write(0, 0, '姓名')
worksheet.write(0, 1, '年龄')
worksheet.write(1, 0, '张三')
worksheet.write(1, 1, 18)

# 保存Excel文件
workbook.close()

3. 处理数据

在创建了Excel文件后，我们需要将数据写入Excel文件中。我们可以在内存中生成数据，也可以从数据库或其他数据源中获取数据。

内存中生成数据

在内存中生成数据非常简单，我们只需要按照一定规则生成数据即可。下面的代码展示了如何内存中生成数据，并将数据写入Excel文件中。

import xlwt

# 创建Excel文件
workbook = xlwt.Workbook()

# 添加工作表
worksheet = workbook.add_sheet('Sheet1')

# 生成数据
data = [[1, '张三', 18], [2, '李四', 20], [3, '王五', 22]]

# 将数据写入Excel文件
for i, row in enumerate(data):
    for j, col in enumerate(row):
        worksheet.write(i, j, col)

# 保存Excel文件
workbook.save('test.xls')

从数据库获取数据

在实际项目中，通常需要从数据库中获取数据，并将数据写入Excel文件中。下面的代码展示了如何从MySQL数据库中获取数据，并将数据写入Excel文件中。

import xlwt
import pymysql

# 创建Excel文件
workbook = xlwt.Workbook()

# 添加工作表
worksheet = workbook.add_sheet('Sheet1')

# 获取数据库连接
conn = pymysql.connect(host='localhost',
                       port=3306,
                       user='root',
                       password='password',
                       database='test')

# 获取游标
cursor = conn.cursor()

# 执行查询语句
sql = 'SELECT * FROM user'
cursor.execute(sql)

# 获取结果
data = cursor.fetchall()

# 将数据写入Excel文件
for i, row in enumerate(data):
    for j, col in enumerate(row):
        worksheet.write(i, j, col)

# 关闭游标和连接
cursor.close()
conn.close()

# 保存Excel文件
workbook.save('test.xls')

以上就是使用xlwt和xlsxwriter库生成Excel文件的示例，希望能对你有所帮助。