7.4.6 批量加载
要对一个很大的数据集应用相同的SQL指令,可以使用executemany()。这个方法对于加载数据很有用,因为这样可以避免在Python中忙处理输入,而允许底层的库应用循环优化。
下面这个示例程序使用csv模块从一个逗号分隔值文件读取任务列表,并把它们加载到数据库。
import csv
import sqlite3
import sys
db_filename = 'todo.db'
data_filename = sys.argv[1]
SQL = """
insert into task (details,priority,status,deadline,project)
values (:details,:priority,'active',:deadline,:project)
"""
with open(data_filename,'rt') as csv_file:
csv_reader = csv.DictReader(csv_file)
with sqlite3.connect(db_filename) as conn:
cursor = conn.cursor()
cursor.executemany(SQL,csv_reader)
文件task.csv包含一下示例数据。
运行结果: