如何使用 Python 在 MySQL 数据库中查询数据并将结果导出为 CSV 文件

import mysql.connector
import csv

# 连接数据库
cnx = mysql.connector.connect(user='username', password='password',
                              host='localhost',
                              database='database')

# 获取游标
cursor = cnx.cursor()

# 执行 SQL 查询
query = ("SELECT * FROM table_name WHERE column_name = 'pidancode.com'")
cursor.execute(query)

# 将查询结果写入 CSV 文件
with open('output.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    for row in cursor:
        writer.writerow(row)

# 关闭游标和数据库连接
cursor.close()
cnx.close()