##連接數據庫,查詢結果寫入數據到excel
import pymysql
import xlwt
def get_sel_excel(file_excel):
#建立連接
conn = pymysql.connect('localhost','root','zhaojing1314','stu')
#建立遊標
cursor = conn.cursor()
sel_sql = "SELECT * FROM PYTHON;"
print("開始查詢表!")
#執行sql語句
cursor.execute(sel_sql)
#獲取查詢到結果
res = cursor.fetchall()
print(res)
w_excel(res)
#操作excel
def w_excel(res):
book = xlwt.Workbook() #新建一個excel
sheet = book.add_sheet('STUDENTS6') #新建一個sheet頁
title = ['NUM','STU_NUM','NAME','CLASS']
#寫表頭
i = 0
for header in title:
sheet.write(0,i,header)
i+=1
#寫入數據
for row in range(1,len(res)):
for col in range(0,len(res[row])):
sheet.write(row,col,res[row - 1][col])#這裏不寫row-1則會把第一個數據丟失
row+=1
col+=1
book.save('STUDENTS6')
print("導出成功!")
if __name__ == "__main__":
file_excel = r"D:\python\python名單.xls"#這條語句可以不寫
get_sel_excel(file_excel)