用Python 写一个批量导入数据到SQLite3
时间: 2023-05-20 21:06:40 浏览: 95
可以使用 Python 的 sqlite3 模块来实现批量导入数据到 SQLite3 数据库。以下是一个示例代码:
```python
import sqlite3
import csv
# 连接到 SQLite3 数据库
conn = sqlite3.connect('mydatabase.db')
cursor = conn.cursor()
# 创建表格
cursor.execute('''CREATE TABLE IF NOT EXISTS mytable
(id INTEGER PRIMARY KEY, name TEXT, age INTEGER)''')
# 打开 CSV 文件并读取数据
with open('data.csv', 'r') as csvfile:
reader = csv.reader(csvfile)
next(reader) # 跳过标题行
for row in reader:
# 插入数据到表格
cursor.execute("INSERT INTO mytable (name, age) VALUES (?, ?)", (row[0], row[1]))
# 提交更改并关闭连接
conn.commit()
conn.close()
```
这个代码会创建一个名为 `mydatabase.db` 的 SQLite3 数据库,并在其中创建一个名为 `mytable` 的表格。然后,它会从一个名为 `data.csv` 的 CSV 文件中读取数据,并将数据批量导入到 `mytable` 表格中。注意,这个代码假设 CSV 文件的第一列是名字,第二列是年龄。如果你的 CSV 文件有不同的列名或列顺序,需要相应地修改代码。