psycopg2 dataframe入库(pgsql数据库)

最新推荐文章于 2024-08-01 20:33:10 发布

tgn123456

最新推荐文章于 2024-08-01 20:33:10 发布

阅读量376

点赞数

文章标签：数据库 python

本文链接：https://blog.youkuaiyun.com/RaymondCoder/article/details/132970368

版权

本文介绍了如何使用Python的pandasDataFrame和psycopg2库将DataFrame数据导出到PostgreSQL数据库中，包括DataFrame转换为CSV字符串、连接数据库、创建游标并使用`copy_from`方法批量导入数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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