利用python查询mysql数据库表中数据并生成为csv文件

需求：从mysql数据库中查询某表的数据，并将表中数据生成csv文件保存在指定文件下。

话不多说，直接上代码：

import pymysql
from settings import MYSQL_HOST, MYSQL_DB, MYSQL_PWD, MYSQL_USER
import csv

def mysql_db_sz_stock():
    # 连接数据库
    connect = pymysql.connect(host=MYSQL_HOST,   # 本地数据库
                              user=MYSQL_USER,
                              password=MYSQL_PWD,
                              db=MYSQL_DB,
                              charset='utf8')  # 服务器名,账户,密码，数据库名称
    cursor = connect.cursor()

    sql = "select * from daily_view"

    cursor.execute(sql)
    data = cursor.fetchall()
    connect.commit()  # 提交到数据库执行
    print("查询数据成功")

    #  ---------------------关闭数据库
    cursor.close()  # 关闭游标
    connect.close()  # 关闭数据库连接
    return data

# 将数据保存为csv文件
def write_csv():
    data = mysql_db_sz_stock()
    filename = './csv/daily_view.csv'  # 文件名和路径
    with open(filename, mode='w', encoding='utf-8') as f:
        write = csv.writer(f, dialect='excel')
        write.writerow(['symbol', 'datetime', 'Open', 'High', 'Low', 'Close', 'Volume'])  # 先写下标题
        for item in data:
            write.writerow(item)

write_csv()

运行后发现csv文件中数据隔行显示

解决办法：添加newline

def write_csv():
    data = mysql_db_sz_stock()
    filename = './csv/daily_view.csv'  # 文件名和路径
    with open(filename, mode='w', newline='', encoding='utf-8') as f:  # 添加newline=''
        write = csv.writer(f, dialect='excel')
        write.writerow(['symbol', 'datetime', 'Open', 'High', 'Low', 'Close', 'Volume'])  # 先写下标题
        for item in data:
            write.writerow(item)

问题解决。