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 in 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