首页 > 解决方案 > 如何在 pd.io.sql.get_schema 中指定 sql 的风格?

问题描述

我正在尝试使用pd.io.sql.get_schema从数据框生成 postgres 模式。

没有文档,pd.io.sql.get_schema但是从这个(https://github.com/pandas-dev/pandas/issues/9960)链接说我可以指定一种sql。

但是,此功能似乎已被弃用,而是我可以指定一个类似 postgresql 的引擎(从 Pandas Dataframe 生成 SQL 语句)。我该怎么做呢?

到目前为止,这是我的代码:

pd.io.sql.get_schema(df.reset_index(), 'data')

对生成模式的所有建议开放。

标签: python-3.xpandasschema

解决方案


我相信您使用 SQLAlchemy 创建到 postgreSQL 数据库的连接,然后将该连接传递给conkwarg。例如:

import numpy as np
import pandas as pd
import sqlalchemy

dates = pd.date_range('20130101',periods=6)
df = pd.DataFrame(np.random.randn(6,4),index=dates,columns=list('ABCD'))

url = 'postgresql://USER:PASSWORD@HOST:PORT/DATABASE'
con = sqlalchemy.create_engine(url, client_encoding='utf8')
print(pd.io.sql.get_schema(df.reset_index(), 'data', con=con))
CREATE TABLE data (
        index TIMESTAMP WITHOUT TIME ZONE,
        "A" FLOAT(53),
        "B" FLOAT(53),
        "C" FLOAT(53),
        "D" FLOAT(53)
)

推荐阅读