使用Python從 MySQL寫數據到Excel

直接上代碼:

#!/usr/bin/env python
#coding:utf-8

import xlwt
import MySQLdb
import datetime

database = MySQLdb.connect(host='192.168.1.30',user='root',passwd='123456',db='crm')
#設置字符集
database.set_character_set('utf8')
cursor = database.cursor()
cursor.execute('SET NAMES utf8;') 
cursor.execute('SET CHARACTER SET utf8;')
cursor.execute('SET character_set_connection=utf8;')

starttime = datetime.datetime.now()
print '開始時間:%s' % (starttime)

#通過SQL得到該表有多少行,如果想取出指定的數據,只需要在後面加where條件即可。
sql2 = 'select count(*) from bill_test;';
cursor.execute(sql2)
count_rows=cursor.fetchone()[0]

wbk = xlwt.Workbook(encoding='utf-8',style_compression=0)
sheet = wbk.add_sheet('sheet 1', cell_overwrite_ok=True)
#設置寫excel的樣式
style = xlwt.XFStyle()
font = xlwt.Font()
font.name = 'Times New Roman'
#0x0190設置字體爲20,默認爲0x00C8 字體爲10 ,0x00C8爲十六進制的數字
font.height = 0x0190
font.bold = True
style.font = font
#查詢得到該表有多少列
query_colums="select count(*) from information_schema.COLUMNS where TABLE_SCHEMA='crm' and table_name='bill_test';"
cursor.execute(query_colums)
count_cols = cursor.fetchone()[0]

sql = 'select member_id, name, tel, phone, dq_datetime, address, parking from bill_test;'
cursor.execute(sql)

#定義所有的列名,共7列
columnName = ['賬號','名稱','電話','手機','到期日期','地址','園區名稱']  
#將列名插入表格,共7列
for i in range(len(columnName)):         
        sheet.write(0,i,columnName[i],style)

#通過循環取出每一行數據,寫入excel    
for i in range(1,count_rows-1):
    data = cursor.fetchone()
    for j in range(0,count_cols-1):
        sheet.write(i,j,data[j],style)       
cursor.close()
database.close()
wbk.save('C:\Users\XUWU\Desktop\data01.xls')   

endtime=datetime.datetime.now()
print '結束時間:%s' % (endtime)
print '用時:%s 秒' % (endtime-starttime)


執行情況:

wKiom1hYzHiCOf4YAAAjP-FFP_g772.png-wh_50


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章