从数据库读取数据转换成dataframe及dataframe写进数据库

 在利用python处理数据库时,难免会碰到需要数据在两个工具之间变换,本文展现了将数据库数据转换成python常用的dataframe格式,并将dataframe格式写入数据库的方法。

import sqlalchemy as sqla
import pymysql
import pandas as pd

#定义从数据库读取数据转换成dataframe函数
def sql_query(sql, db):
    conn = pymysql.connect(host='120.24.38.6', user='username', passwd='password', db=db, port=3308, charset='utf8')
    cursor = conn.cursor()
    cursor.execute(sql)
    results = cursor.fetchall()
    columnDes = cursor.description #获取连接对象的描述信息
    cursor.close()
    conn.close()
    columnNames = [columnDes[i][0] for i in range(len(columnDes))]
    results = pd.DataFrame([list(i) for i in results],columns=columnNames)
    return results

#编写sql语句并保存执行结果
sql2 = """select * from table'"""
result1 = sql_query(sql2, db)

#将dataframe数据写入数据库
db = sqla.create_engine('mysql+pymysql://gongxiang:[email protected]:3308/data_analysis')
table3.to_sql('overseas_warehouse', con=db, index=False, if_exists='replace')

 

你可能感兴趣的:(mysql,python)