python - 如何将数据类型从 python pandas 映射到 postgres 表？

Question

我有一个 pandas.DataFrame，其中的列具有不同的数据类型，例如 object、int64 等。我有一个使用适当数据类型创建的 postgresql 表。我想将所有数据框数据插入到 postgresql 表中。应该如何做到这一点？

Note : The data in pandas is coming from another source so thedata types are not specified manually by me.

Answer 1

最简单的方法是使用sqlalchemy:

from sqlalchemy import create_engine

engine = create_engine('postgresql://abc:def@localhost:5432/database')
df.to_sql('table_name', engine, if_exists='replace')

如果表存在，你可以选择你想用if_exists选项做什么

if_exists {‘fail’, ‘replace’, ‘append’}, default ‘fail’

如果该表不存在，它将创建一个具有相应数据类型的新表。