[Python]DataFrame读取与写回数据库数据

一、简述
数据分析中经常需要用Python分析处理数据库的数据(尤其是用Pandas的DataFrame), 然后把处理和分析的结果写回到数据库中指定的表中，以下为解决此需求的代码（以SQL SERVER为例)

二、代码

1.导入必要的库

import pandas as pd
import pymssql
from sqlalchemy import create_engine

2.读取数据库数据到DataFrame

conn = pymssql.connect(
    host='服务器的地址',
    user='数据库用户名',
    password='数据库密码',
    database='数据库名称',
    charset='utf8'
    )
cursor = conn.cursor(as_dict=True)
#此处如不填as_dict参数则返回结果中的"Value"将默认为元组形式
cursor.execute(
    """在此插入你的SQL查询语句"""
    )
    
results = cursor.fetchall()
results_df = pd.DataFrame(data = results)
#数据至此读取到名为"results_df"的DataFrame中

3.把处理好的DataFrame数据写入到数据库

results_df_copy = results_df.copy()
#假设数据已处理好并复制到results_df_copy
engine = create_engine(
    'mssql+pymssql://数据库用户名:数据库密码@服务器地址/数据库名称'
    )
results_df_copy.to_sql('要写入的目标表的表名',engine)
conn.close()

End~