如何将DataFrame写入postgres表？

Question

有DataFrame.to_sql方法，但仅适用于mysql，sqlite和oracle数据库。我无法传递给此方法postgres连接或sqlalchemy引擎。

Answer 1

这就是我做的。

可能更快，因为它正在使用execute_batch：

# df is the dataframe

if len(df) > 0:

    df_columns = list(df)

    # create (col1,col2,...)

    columns = ",".join(df_columns)

    # create VALUES('%s', '%s",...) one '%s' per column

    values = "VALUES({})".format(",".join(["%s" for _ in df_columns])) 

    #create INSERT INTO table (columns) VALUES('%s',...)

    insert_stmt = "INSERT INTO {} ({}) {}".format(table,columns,values)

    cur = conn.cursor()

    psycopg2.extras.execute_batch(cur, insert_stmt, df.values)

    conn.commit()

    cur.close()