python：连接mysql获取带列名的数据，格式化 mysql 结果集

雨点打透心脏的1/2处 2023-10-05 11:44 5阅读 0赞

格式化数据结果如下，为字典数组

[{
        'id': 1, 'name': 'admin', 'password': '123456'}, 
    
    {
        'id': 5, 'name': 'oppo', 'password': '456789'}, 
    
    {
        'id': 6, 'name': 'vivo', 'password': '987654'}]

实现代码如下

import pymysql
    
    conn = pymysql.connect(host='127.0.0.1', user='root', password='123456', database='python_test', charset='utf8')
    cursor = conn.cursor()
    
    def get_result():
        sql = "select * from user"
        cursor.execute(sql)
        result = cursor.fetchall()
        fields = cursor.description
        res = format_data(fields, result)
        cursor.close()
        conn.close()
        print(res)
    
    # 数据格式化 fields 字段名，result 结果集
    def format_data(fields, result):
        # 字段数组 ['id', 'name', 'password']
        field = []
        for i in fields:
            field.append(i[0])
        # 返回的数组集合 形式[{'id': 1, 'name': 'admin', 'password': '123456'}]
        res = []
        for iter in result:
            line_data = {
        }
            for index in range(0, len(field)):
                line_data[field[index]] = iter[index]
            res.append(line_data)
        return res
    
    get_result()

下面做一简单测试

mysql user表数据如下  
![在这里插入图片描述][20201015213507697.png_pic_center]  
运行结果如下

[{
        'id': 1, 'name': 'admin', 'password': '123456'}, 
    
    {
        'id': 5, 'name': 'oppo', 'password': '456789'}, 
    
    {
        'id': 6, 'name': 'vivo', 'password': '987654'},
    
     {
        'id': 7, 'name': 'huawei', 'password': ''}]

python 有没有现成的方法或者模块，初涉python暂未发现，知道的朋友知会一声。

[20201015213507697.png_pic_center]: https://img-blog.csdnimg.cn/20201015213507697.png#pic_center