如何在 python 中定义正在写入 AWS Redshift Postgres DB 的数据的列类型

Question

我有一个看起来像这样的熊猫数据框：

A       B                       C         D
80.00   2020-03-07 19:36:50     testing   approved
80.00   2020-03-07 19:46:50     testing   approved

我已经成功地将这个 pandas 数据框写入数据库，如下所示：

conn = create_engine('postgresql://username:password@redshift-blablabla.eu-west-1.redshift.amazonaws.com:5439/database_name')

dataframe.to_sql('table_name', conn,index=False, if_exists='replace')

当我尝试将此表用作在 Superset 中可视化数据的源时，我注意到所有内容都存储为 varchar，我不能使用列 B 作为日期戳来序列化数据。

写入数据库时​​如何定义列类型？并且日期戳是否应该只有日期而没有小时才能作为日期戳有效？

感谢您的帮助。

Answer 1

to_sql 有一个参数dtype，您可以通过该参数将单个列的数据类型作为字典提供，或者同时将所有列的数据类型作为标量提供。你试过了吗？

dtype : 字典或标量，可选

指定列的数据类型。如果使用字典，键应该是列名，值应该是 SQLAlchemy 类型或 sqlite3 旧模式的字符串。如果提供了标量，它将应用于所有列。