从python API将sql数据写入csv

Question

我正在从 python API 运行 SQL 查询，并希望收集结构化数据（在他们自己的标题下的列数据）.CSV 格式。

这是我目前的代码。

import pymysql.cursors
import csv


conn = pymysql.connect(host='159.XXX.XXX.XXX',user='proXXX',password='PXX',db='pXX',charset='utf8mb4',cursorclass=pymysql.cursors.DictCursor)
cursor = conn.cursor()
print (type(conn))
sql = "SELECT id,author From researches WHERE id < 20 " 
cursor.execute(sql)
data = cursor.fetchall()
print (data)
with open('metadata.csv', 'w', newline='') as f_handle:
    writer = csv.writer(f_handle,delimiter=',')
    header = ['id', 'author']
    writer.writerow(header)
    for row in data:
        writer.writerow(row)

现在数据正在控制台上打印，但没有进入 .CSV 文件，这就是我得到的 asnoutput。我错过了什么？请帮忙。

Answer 1

with open('metadata.csv', 'w', newline='') as f_handle:
    fieldnames = ['id', 'author']
    writer = csv.DictWriter(f_handle, fieldnames=fieldnames)
    writer.writeheader()
    for row in data:
        writer.writerow(row)

所以问题是，您的数据是字典的形式，而 Writer 对象需要元组。您应该改用 DictWriter 对象。