您的位置:首页 > 编程语言 > Python开发

Python实现数据库数据导入Excel

2018-04-04 14:41 811 查看

1 任务简介

把PostgreSql数据库中5张表的数据导出到Excel文件中,查询的字段涉及到5张表,所以需要联合查询,其中还有两张表存在自引用,所以还会更复杂一些。

2 工具

使用
psycopg2
模块连接数据库使用
xlwt
模块操作Excel文件

2.1 psycopg2

这个模块用法很常规,连接数据库执行SQL获取结果重点在于SQL语句的构造,因为涉及多个表,所以不得已要使用
join
,那么为了考虑效率应当把数据量大的表放在小表的前面。

2.2 xlwt

这个模块用于写Excel文件,与之对应的还有
xlrd
来完成读文件的操作。 用法也很简单,新建Excel文件新建sheet表定位起始单元格写内容保存&关闭文件需要注意的问题有两点:文件编码,新建文件时需要指明
utf-8
表格大小,由于xls文件最多只能到65535行,所以表过大的时候要么换
xlsx
格式,要么切分成多个表。Python处理
xlsx
的模块是
xlsxwriter

3 代码

具体实现如下:
#!/usr/bin/env python
# -*- coding: utf-8 -*-

import psycopg2
import xlwt

hostIp = '127.0.0.1'
port = '5432'
dbName = 'source'
user = 'liuyueyi'
password = 'password'

sql = '''
select site.name,study.name,sheets.id,sheets.patient_name,crf.label,section.label,form_rows.uid,form_rows.label,sheet_row_values.value
from sheet_row_values
inner join form_rows on form_rows.uid = sheet_row_values.uid
inner join form_groups section on section.id = form_rows.group_id
inner join form_groups crf on section.parent_id = crf.id
inner join sheets on sheets.id = sheet_row_values.sheet_id
inner join departments site on sheets.center_id = site.id
inner join departments study on site.parent_id = study.id;
'''

file = xlwt.Workbook(encoding='utf-8')

sheet1 = file.add_sheet(u'sheet1')

row0 = [u'site_name',u'study_name',u'sheets_id',u'patient_name',u'crf_name',u'section_name',u'form_rows_uid',u'form_rows_label',u'row_value']
for i in range(0, len(row0)):
sheet1.write(0,i,row0[i])

conn = psycopg2.connect(database=dbName,host=hostIp,port=port,user=user,password=password)
cur = conn.cursor()
cur.execute(sql)
rows = cur.fetchall()

for i i
4000
n range(0,len(rows)):
for j in range(0,len(rows[i])):
sheet1.write(i+1,j,rows[i][j])
file.save('result.xls')
原文链接:https://blog.csdn.net/liuyueyi1995/article/details/60587916
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: