Python Pandas to_sql，如何用主键创建表？

  • Z时代
  • 2024-01-10
  • 分类：问答

我想用Pandas的to_sql函数创建一个具有主键的MySQL表（在mysql表中具有主键通常是一种好习惯），如下所示：

group_export.to_sql(con = db, name = config.table_group_export, if_exists = 'replace', flavor = 'mysql', index = False)

但这会创建一个没有任何主键（甚至没有任何索引）的表。

该文档提到了参数“ index_label”，该参数与“ index”参数结合可用于创建索引，但未提及主键的任何选项。

文献资料

回答：

免责声明：这个答案是实验性的，而不是实用的，但也许值得一提。

我发现该类pandas.io.sql.SQLTable已命名为参数key，如果您为其分配了字段名称，则该字段将成为主键：

不幸的是，您不能只从DataFrame.to_sql()函数传递此参数。要使用它，您应该：

  1. 创建pandas.io.SQLDatabase实例

    engine = sa.create_engine('postgresql:///somedb')

    pandas_sql = pd.io.sql.pandasSQL_builder(engine, schema=None, flavor=None)

  2. 定义函数类似于pandas.io.SQLDatabase.to_sql()但带有附加*kwargs参数的函数，该函数传递给pandas.io.SQLTable在内部创建的对象（我刚刚复制了原始to_sql()方法并添加了*kwargs）：

    def to_sql_k(self, frame, name, if_exists='fail', index=True,
           index_label=None, schema=None, chunksize=None, dtype=None, **kwargs):
    if dtype is not None:
        from sqlalchemy.types import to_instance, TypeEngine
        for col, my_type in dtype.items():
            if not isinstance(to_instance(my_type), TypeEngine):
                raise ValueError('The type of %s is not a SQLAlchemy '
                                 'type ' % col)
    table = pd.io.sql.SQLTable(name, self, frame=frame, index=index,
                     if_exists=if_exists, index_label=index_label,
                     schema=schema, dtype=dtype, **kwargs)
    table.create()
    table.insert(chunksize)

  3. 使用SQLDatabase实例和要保存的数据框调用此函数

    to_sql_k(pandas_sql, df2save, 'tmp',
        index=True, index_label='id', keys='id', if_exists='replace')

我们得到类似

CREATE TABLE public.tmp
(
  id bigint NOT NULL DEFAULT nextval('tmp_id_seq'::regclass),
...
)

在数据库中。

PS当然，您可以安装monkey-patch DataFrame，io.SQLDatabase并io.to_sql()可以方便地使用此替代方法。

以上是 Python Pandas to_sql，如何用主键创建表？ 的全部内容， 来源链接： utcz.com/qa/435557.html

回到顶部