使用 StringIO 將 pandas dataframe 保存到 postgres sql 表

Question

我正在嘗試將 dataframe 持久保存到 postgres 表中，但沒有運氣。

I have a table table1 with fields f1,f2, f3

df = pd.DataFrame({"f1":["A", "B"],
                   "f2": [1, 2],
                   "f3": ["p", "f"]
            })
buffer = StringIO()
df.to_csv(buffer, sep=",", index=False, header=False)
buffer.seek(0)

# conn is a sqlalchemy.engine.base.Connection object
cur =  conn.connection.cursor(cursor_factory=psycopg2.extras.DictCursor)
# To set the schema
cur.execute(f"SET search_path TO my_schema")
cur.copy_from(buffer,'table1', sep=",")
cur.close()

不會拋出錯誤，但數據不會寫入 db 表。 束手無策。

Answer 1

嘗試在 cur.close conn.connection.commit() cur.close() ：

cur.execute(f"SET search_path TO my_schema")
cur.copy_from(buffer,'table1', sep=",")
cur.close()

# Add this line
conn.connection.commit()