pymysql 下载数据表存放为csv

步骤

下载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()
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容