问Pandas to_sql，如何创建一个带有主键的表？

EN

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

我发现类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 _ is =‘fail’，index=True，index_label=None，schema=None，chunksize=None，dtype=None，**kwargs)：如果dtype不是None:从sqlalchemy.types import to_instance，TypeEngine for dtype.items()，my_type in dtype.items()：如果不是isinstance(to_instance(my_type)，TypeEngine)：self (‘类型%s不是SQLAlchemy’‘类型’%col‘)表=pd.io.sql.SQLTable(名称、自我、frame=frame、index=index、if_exists=if_exists、index_label=index_label、schema=schema、dtype=dtype，**kwargs) table.create() table.insert(块状)
3. 使用您的SQLDatabase实例和您想要保存的数据调用此函数 to_sql_k(pandas_sql，df2save，'tmp'，index=True，index_label='id'，key=‘id’，if_if=‘index=True’)

我们得到了类似于

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

在数据库里。

当然，您可以使用猴子补丁DataFrame、io.SQLDatabase和io.to_sql()函数来方便地使用这个解决方法。

只需添加主键后，上传表与熊猫。

group_export.to_sql(con=engine, name=example_table, if_exists='replace', 
                    flavor='mysql', index=False)

with engine.connect() as con:
    con.execute('ALTER TABLE `example_table` ADD PRIMARY KEY (`ID_column`);')

对于大熊猫0.15，至少对于某些口味，您可以使用参数dtype来定义主键列。您甚至可以以这种方式激活AUTOINCREMENT。对于sqlite3，如下所示：

import sqlite3
import pandas as pd

df = pd.DataFrame({'MyID': [1, 2, 3], 'Data': [3, 2, 6]})
with sqlite3.connect('foo.db') as con:
    df.to_sql('df', con=con, dtype={'MyID': 'INTEGER PRIMARY KEY AUTOINCREMENT'})