python pandas写入数据后保存_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 = pymysql.connect(host, user, password, db)

cur = con.cursor() # 获取操作游标，也就是开始操作

cur.execute(sql) # 输入要查询的SQL

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))

df = pd.DataFrame(data=data1,columns=columns) # mysql查询的结果为元组，需要转换为列表

df.to_csv(csv_path,index=None)

if __name__ == '__main__':

sql = 'select * from table;'

rel,columns = getrel(sql)

getcsv(rel,columns)