如何使用 Python 将 MySQL 查询到字典中？

Question

我正在尝试将我的查询结果提取到 Python 中的字典中，但不太确定如何执行此操作。

我在光标配置中设置了 Dictionary=true 但这似乎没有按预期工作。

import mysql.connector as mariadb
#  DB VARIABLES
mariadb_connection = mariadb.connect(user='user', password='pass', database='mydb')
cursor = mariadb_connection.cursor(dictionary=True)

cursor.execute("SELECT id, name FROM songs")
rows = cursor.fetchall()
print(rows)
if isinstance(rows, dict):
    print("this IS a dictionary")

上面的代码不打印任何东西。

Answer 1

当您使用.fetchall() 时，您将检索多个结果，即使查询只有一个结果，它也会是list of all results。

---

使用dictionary=True 将为您提供dict 的列表，其字段名称如下

[{'id': '123456', 'name': 'This is the supe song 1'},
 {'id': '456789', 'name': 'This is the supe song 2'},
 {'id': '789123', 'name': 'This is the supe song 3'}]

没有dictionary=True，你只会得到一个元组形式的值列表

[('123456', 'This is the supe song 1'),
 ('456789', 'This is the supe song 2'),
 ('789123', 'This is the supe song 3')]

---

下面是True所以

isinstance(rows, list) # True
isinstance(rows[0], dict) # True

Answer 2

如果您想获取字典，您需要决定将什么用作字典键。假设是第一个返回的字段，即id：

mariadb_connection = mariadb.connect(user='user', password='pass', database='mydb')
cursor = mariadb_connection.cursor()
cursor.execute("SELECT id, name FROM songs")
rows = cursor.fetchall()
dic = { t[0]: t[1:] for t in rows }

当然，要使其正常工作，id 中的值必须是唯一的。