步骤
下载mysql 数据,然后把字段 读取出来。
根据pandas 的dataframe 数据格式 用列表推导式 整理数据
最后 导出 为csv
代码
# -*- coding: UTF-8 -*-
import pymysql
import pandas as pd
from config_mysql import *
def download(sql,file):
mysql = pymysql.connect(host=host, user=user, password=password, port=port, db=db)
cursor = mysql.cursor()
import time
start_t = time.time()
cursor.execute(sql)
results = cursor.fetchall() # 用于返回多条数据
print('spend {}s'.format(int(time.time() - start_t)))
df = pd.DataFrame(results, columns=[u[0] for u in cursor.description])
print(len(df))
df.to_csv(file, encoding='utf-8')
# 3.关闭查询
cursor.close()
# 4.关闭数据库
mysql.close()