Python如何将csv写入sqlite
要将CSV文件内容导入SQLite数据库,您可以使用Python的sqlite3
模块和csv
模块。以下是一个简单的例子:
import csv
import sqlite3
# 创建或连接到SQLite数据库
conn = sqlite3.connect('example.db')
cursor = conn.cursor()
# 创建表
cursor.execute('''
CREATE TABLE IF NOT EXISTS csv_data (
id INTEGER PRIMARY KEY,
column1 TEXT,
column2 TEXT,
column3 TEXT
)
''')
# 打开CSV文件
with open('data.csv', newline='') as csvfile:
csv_reader = csv.reader(csvfile, delimiter=',')
# 跳过标题行
next(csv_reader)
# 插入数据
for row in csv_reader:
cursor.execute('''
INSERT INTO csv_data (column1, column2, column3)
VALUES (?, ?, ?)
''', (row[0], row[1], row[2]))
# 提交更改并关闭连接
conn.commit()
conn.close()
确保替换example.db
为您的数据库文件名,data.csv
为您的CSV文件名,以及column1
, column2
, column3
等为CSV文件中的列名。如果CSV文件有标题行,请保证在读取CSV时跳过它。如果CSV文件没有标题行,请确保在插入数据时正确引用列索引。
评论已关闭