以下代码是将mysql数据库数据导入到csv文件
import pymysql
import csv
def from_mysql_get_all_info():
conn = pymysql.connect(
host='xxxxxx',
port=3306,
user='xxxx',
db=xx',
password='xxxxx',
charset='utf8mb4'
cursor = conn.cursor()
sql = 'select * from py.tencent_news'
cursor.execute(sql.encode('utf-8'))
data = cursor.fetchall()
conn.close()
return data
def write_csv():
data = from_mysql_get_all_info()
filename = 'data/corpus.csv'
with open(filename,mode='w',encoding='utf-8') as f:
write = csv.writer(f,dialect='excel')
for item in data:
write.writerow(item)
write_csv()