python读取MySQL数据使⽤pandas写⼊到csv,并保存列名实现的功能:通过pymysql连接MySQL数据库,查询所需数据,然后使⽤pandas写⼊到csv⽂件,并且写⼊包含数据列名。
import sys
import pandas as pd
import pymysql
import time
csv_path=sys.argv[1]
host=sys.argv[2]
user=sys.argv[3]
password=sys.argv[4]
db=sys.argv[5]
# host='59.202.45.105'
# user='lpz_test'
# password='lpz_test'
# db='test'
def getrel(sql):
try:
con = t(host, user, password, db)
cur = con.cursor() # 获取操作游标,也就是开始操作
col_result = cur.description # 获取查询结果的字段描述
rel = cur.fetchall()
# 获取字段名,以列表形式保存
columns = []
for i in range(0, len(col_result)):
columns.append(col_result[i][0])
# print(columns)
cur.close()
con.close()
return rel,columns
except Exception as e:
print(e)
print("请检查数据库连接信息")
exit(-1)
def getcsv(rel,columns):
data1 = list(map(list, rel))python怎么读csv数据
df = pd.DataFrame(data=data1,columns=columns) # mysql查询的结果为元组,需要转换为列表
<_csv(csv_path,index=None)
if __name__ == '__main__':
sql = 'select * from table;'
rel,columns = getrel(sql)
getcsv(rel,columns)
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198,我们将在24小时内删除。
发表评论