Python，将CSV文件转换为SQL表

Question

我有一个没有标题的 CSV 文件，我正在尝试从文件中的某些列创建一个 SQL 表。我尝试了这里给出的解决方案:Importing a CSV file into a sqlite3 database table using Python ,但不断收到未定义 col1 的错误。然后我尝试在我的 CSV 文件中插入 header ，但仍然收到 KeyError。

感谢任何帮助! (我对SQL根本不是很熟悉)

Answer 1

如果 .csv 文件没有标题，你不想使用 DictReader； DictReader 假定第 1 行是一组标题，并将它们用作每个后续行的键。这可能就是您收到 KeyError 的原因。

该链接示例的修改版本:

import csv, sqlite3

con = sqlite3.connect(":memory:")
cur = con.cursor()
cur.execute("CREATE TABLE t (col1, col2);")

with open('data.csv','rb') as fin:
    dr = csv.reader(fin)
    dicts = ({'col1': line[0], 'col2': line[1]} for line in dr)
    to_db = ((i['col1'], i['col2']) for i in dicts)

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