1. 程式人生 > >python實現查詢的資料寫入到excel

python實現查詢的資料寫入到excel

#coding=utf-8
import sys
import xlwt
import pymysql as MySQLdb #這裡是python3 如果你是python2.x的話,import MySQLdb
import datetime


host = '192.168.10.109'
user = 'root'
pwd = ''
port = 3306
db = 'com66nao_mi'
sheet_name = 'report'
out_path = r'D:\SQL\aaa'+'.xls'
print(out_path)
sql = '''select * from mi_orgs;'''

def export():
conn = MySQLdb.connect(host,user,pwd,db,charset='utf8') 
cursor = conn.cursor() 
count = cursor.execute(sql) 
print("查詢出" + str(count) + "條記錄")

#來重置遊標的位置 
cursor.scroll(0,mode='absolute')
#搜取所有結果 
results = cursor.fetchall()

# 獲取MYSQL裡面的資料欄位名稱
fields = cursor.description
workbook = xlwt.Workbook() # workbook是sheet賴以生存的載體。 
sheet = workbook.add_sheet(sheet_name,cell_overwrite_ok=True)

# 寫上欄位資訊 
for field in range(0,len(fields)): 
sheet.write(0,field,fields[field][0])

# 獲取並寫入資料段資訊 
row = 1 
col = 0 
for row in range(1,len(results)+1): 
for col in range(0,len(fields)): 
sheet.write(row,col,u'%s'%results[row-1][col]) 

workbook.save(out_path)

#結果測試
if __name__=="__main__":
export()