psycopg2 dataframe入库(pgsql数据库)

最新推荐文章于 2024-04-10 15:34:46 发布

tgn123456

最新推荐文章于 2024-04-10 15:34:46 发布

阅读量212

点赞数

文章标签：数据库 python

本文链接：https://blog.csdn.net/RaymondCoder/article/details/132970368

版权

def todb(df, tablename, columns=None):
    '''
     df:dataframe
     tablename: 表名
     columns: 列名(dataframe如完全按照表字段 可不写)
    '''
	strio = StringIO()
    df.to_csv(strio, sep='\t', index=False, header=False)
    output = strio.getvalue()
    conn = psycopg2.connect(database="XXX", user="XXX", password="XXX", host="XXX")
    cur = conn.cursor()
    cur.copy_from(StringIO(output), tablename, columns=columns)
    conn.commit()
    cur.close()
    conn.close()
    strio.close()