1 import pandas as pd 2 import numpy as np 3 import matplotlib as mpl 4 import matplotlib.pyplot as plt 5 mpl.rcParams['font.sans-serif']=['KaiTi'] 6 mpl.rcParams['font.serif']=['KaiTi'] #導入需要用到的常用庫 7 import pymysql #導入mysql的連接模塊 8 conn=pymysql.connect(user='root',password='xxxxxx',database='xxx') #設置連接用戶名密碼數據庫等 9 cursor=conn.cursor() #設置游標 設置了才能正常查詢 10 11 def mysql(sql): 12 cursor.execute(sql) #執行查詢語句 13 jieguo=cursor.fetchall() #查看全部查詢結果 14 cols=cursor.description #類似 desc table_name返回結果 15 col=[] #創建一個空列表以存放列名 16 for v in cols: 17 col.append(v[0]) #循環提取列名,並添加到col空列表 18 dfsql=pd.DataFrame(jieguo,columns=col) #將查詢結果轉換成DF結構,並給列重新賦值 19 if dfsql.empty: 20 return 'empty set' #判斷查詢結果為空時返回的值 21 else: 22 return dfsql #以DF結構返回查詢結構,DF.to_excel...導出查詢結果時可以帶列名,這樣就解決了mysql直接導出結果無列名的問題
查詢示例: