我想用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,如果您为其分配了字段名称,则该字段将成为主键:
pandas.io.sql.SQLTable
key
不幸的是,您不能只从DataFrame.to_sql()函数传递此参数。要使用它,您应该:
DataFrame.to_sql()
创建pandas.io.SQLDatabase实例
pandas.io.SQLDatabase
engine = sa.create_engine('postgresql:///somedb')
pandas_sql = pd.io.sql.pandasSQL_builder(engine, schema=None, flavor=None)
定义函数类似于pandas.io.SQLDatabase.to_sql()但带有附加*kwargs参数的函数,该函数传递给pandas.io.SQLTable在内部创建的对象(我刚刚复制了原始to_sql()方法并添加了*kwargs):
pandas.io.SQLDatabase.to_sql()
*kwargs
pandas.io.SQLTable
to_sql()
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)
使用SQLDatabase实例和要保存的数据框调用此函数
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()可以方便地使用此替代方法。
DataFrame
io.SQLDatabase
io.to_sql()