Pandas to_sql,如何用主键创建表
时间: 2023-11-27 11:50:58 浏览: 164
Python 使用Pandas实现数据库的读、写操作 Python源码
5星 · 资源好评率100%
在使用`to_sql`方法将Pandas DataFrame写入数据库时,可以指定表名以及表的主键。如果表不存在,可以通过指定主键来创建新表。
以下是一个示例代码,演示如何使用主键创建表:
```python
import pandas as pd
from sqlalchemy import create_engine
engine = create_engine('sqlite:///mydatabase.db')
df = pd.DataFrame({
'id': [1, 2, 3],
'name': ['Alice', 'Bob', 'Charlie'],
'age': [25, 30, 35]
})
df.to_sql('mytable', engine, if_exists='replace', index=False,
dtype={'id': 'INTEGER PRIMARY KEY', 'name': 'TEXT', 'age': 'INTEGER'})
```
在上面的代码中,我们使用`dtype`参数指定了每个列的数据类型以及`id`列作为主键。如果不存在名为`mytable`的表,它将被创建,并且`id`列将作为主键。
注意,这里使用了`if_exists='replace'`参数,如果表已经存在,将被删除并重新创建。如果不希望删除已存在的表,可以将其设置为`if_exists='append'`,数据将被添加到现有表中。
希望这可以帮助你创建新表并在其中插入数据。
阅读全文