psycopg2 dataframe入库(pgsql数据库)

DataFrametoPostgreSQLwithpsycopg2

最新推荐文章于 2024-12-15 14:50:40 发布

原创最新推荐文章于 2024-12-15 14:50:40 发布 · 433 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据库 #python

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

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