Pandas 在导入 SQLite 数据库时创建关系？

Question

假设我有两个简单的 .csv 文件，我想将它们导入 SQLite 数据库：

id	state
1	State1
2	State2
3	State3

id	district	values	state_fk
1	District1	123	1
2	District2	456	2
3	District3	789	3

我每 2 周收到一次更新版本的 district 文件。我只需读取 csv 文件并通过 df.to_sql('table', con, if_exists='replace') 导入/更新它们。

熊猫有没有办法在states.id和districts.state_fk之间设置PK-FK关系？ 我想用新值定期更新district 表，并且不要在每次更新后再次手动设置关系。

Answer 1

我认为用 pandas 创建外键是不可能的。这不是该模块的用途。

不是每次都替换数据库表，您是否可以先使用 sqlite3 模块清空/截断它，然后使用 if_exists='append' 选项导入数据？看起来像这样：

import sqlite3
con = sqlite3.connect('example.db')
cur = con.cursor()
cur.execute('''TRUNCATE TABLE table''')
con.commit()
con.close()
df.to_sql('table', con, if_exists='append')