使用 Python 将 CSV 文件导入到 sqlite3 数据库表中

Question

我有一个 CSV 文件，我想使用 Python 将该文件批量导入到我的 sqlite3 数据库中。命令是“.import …..”。但它似乎不能这样工作。任何人都可以给我一个如何在 sqlite3 中执行此操作的示例吗？我正在使用 Windows 以防万一。谢谢

原文由 Hossein 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

import csv, sqlite3

con = sqlite3.connect(":memory:") # change to 'sqlite:///your_filename.db'
cur = con.cursor()
cur.execute("CREATE TABLE t (col1, col2);") # use your column names here

with open('data.csv','r') as fin: # `with` statement available in 2.5+
    # csv.DictReader uses first line in file for column headings by default
    dr = csv.DictReader(fin) # comma is default delimiter
    to_db = [(i['col1'], i['col2']) for i in dr]

cur.executemany("INSERT INTO t (col1, col2) VALUES (?, ?);", to_db)
con.commit()
con.close()

原文由 mechanical_meat 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

创建与磁盘上文件的 sqlite 连接留给读者作为练习……但现在 pandas 库使两行代码成为可能

df = pandas.read_csv(csvfile)
df.to_sql(table_name, conn, if_exists='append', index=False)

原文由 Tennessee Leeuwenburg 发布，翻译遵循 CC BY-SA 3.0 许可协议