利用pandas将mysql查询出得结果写入到excel文件

Posted 2020-11-10

#!/usr/bin/env python3

import pandas as pd

import pymysql

#返回SQL结果的函数

def getrel(sql):

    conn =  pymysql.connect(host='localhost',user='root',password='123456',db='test')

    cur = conn.cursor()

    cur.execute('set names utf8')

    cur.execute('select app,name  from tb')  # 输入要查询的SQL

    rel= cur.fetchall()

    cur.close()

    conn.close() 

  return rel

 

#生成xlsx文件的函数

def getxlsx(rel,dt):

    dret = pd.DataFrame.from_records(list(rel))  # mysql查询的结果为元组，需要转换为列表

    dret.to_excel("filename.xlsx",index=False,header=("app","name")）#header 指定列名，index 默认为True，写行名

## xlsx文件默认使用xlsxwriter，可以通过engine="xlsxwriter"指定