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

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

[{'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表数据如下

运行结果如下

[{'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暂未发现，知道的朋友知会一声。