python - 仅从 CSV 导入两列

Question

我有一个包含 20 列和 100 行的 CSV 文件。我想仅使用其中两列创建一个 SQLite 数据库 - 第 1 列中的日期和第 10 列中的温度。我修改了找到的一些示例代码，但不确定如何让它只使用这两列。

import csv, sqlite3
import sys

filename = sys.argv[-1]

con = sqlite3.connect("my.db")
con.execute("create table stats(date, temperature)")                               
cur = con.cursor()

stats = csv.reader(open(filename))

con.executemany("insert into stats(date, temperature) values (?, ?)", stats)

con.commit()

Answer 1

con.executemany("insert into stats(date, temperature) values (?, ?)",
                ((rec[0], rec[9]) for rec in stats))