python:mysql导出数据到excel工具方法

近期肺炎严重,春节期间都在家闭关,闲暇时间捡起了之前学习的python,刚好最近在项目上处理大批量数据,从数据库导出数据效率比较低,就想通过python实现导出,结果效率果然很高,基本5-10秒内就导出了,比从mysql数据库直接导出excel效率高了很多。

需求:实现工具类,实现输入库表或sql语句,从数据库导出10万+级数据,要比从mysql导出快捷方便。

实现:以下是实现方式,基本是迭代方式,先实现导出,再实现大批量导出,再优化效率,然后改成公共方法。

首先我导出的数据的数据量基本都超过10万,所以直接使用python的openpyxl,而不是xlwt,xlwt限制6万多条。

1、先import xlwt库,pip install xlwt安装

2、代码编写思路:通过函数实现导出数据,导出表头无需写死,直接通过指定sql读取。所有封装了2个函数,读数据库封装一个方法,写excel封装另外一个方法,详细代码如下:

import MySQLdb

from openpyxl import Workbook

from openpyxl.compat import range

from openpyxl.utils import get_column_letter

import openpyxl 


def query_all(sql):

    # 连接数据库,查询数据

    db = MySQLdb.connect(host="127.0.0.1", port=3306, user="root", passwd="root", db="test",charset='GBK')

    # 使用cursor()方法获取操作游标

    cur = db.cursor()


    # 使用execute方法执行SQL语句

    cur.execute(sql)  # 返回受影响的行数

    fields = [field[0] for field in cur.description]  # 获取所有字段名

    all_data = cur.fetchall()  # 所有数据

    print(len(all_data))

    return all_data,fields


def read_mysql_to_xlsx(sql,excelname):

    # 循环数据写入内容

    jb_date_lists = query_all(sql)

    #print(jb_date_lists)

    jb_date_list = jb_date_lists[0]

    descripte = jb_date_lists[1]


    #要创建的xlsx名称

    dest_filename = excelname + '.xlsx'

    wb = Workbook()

    ws1 = wb.active

    ws1.title = "data"

    # 列名

    for i in range(0,len(descripte)):

        ws1.cell(row=1,column=i+1,value=descripte[i])


    # 写入数据

    for i in range(2,len(jb_date_list)+2):

        for j in range(0,len(descripte)):

            if jb_date_list[i-2][j] is None:

                ws1.cell(row=i, column=j+1, value='')

            else:

                ws1.cell(row=i, column=j+1, value=jb_date_list[i-2][j])

    # 创建xlsx

    wb.save(filename=dest_filename)


if __name__ == '__main__':

    sql = 'SELECT * FROM user0205 t'

    read_mysql_to_xlsx(sql,'user0205')



以上sql可以直接使用导出数据,亲测OK。效率比较高,5-10秒即可导出10万以上数据。

后期处理数据再也不用mysql客户端导出了,python脚本确实方便。

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容