利用python查询mysql数据库表中数据并生成为csv文件
目录
利用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)问题解决。
