使用Python从 CSV 创建一个 sqlite 数据库

本教程将向您展示如何使用Python从CSV文件创建SQLite数据库。SQLite是一个轻量级的关系型数据库管理系统，非常适合小型项目和嵌入式应用程序。

步骤1：导入模块

首先，必须导入CSV和SQLite模块。CSV模块用于读取CSV文件，SQLite模块用于与SQLite数据库交互。

import csv
import sqlite3

步骤2：连接到数据库

使用以下代码段连接到SQLite数据库：

conn = sqlite3.connect('database.db')

这将打开一个名为“database.db”的数据库文件并创建一个数据库连接对象。

步骤3：创建游标对象

在创建游标对象后，就可以使用它来执行SQL查询和命令。

cursor = conn.cursor()

步骤4：创建表

使用以下代码段创建一个名为“data”的表：

cursor.execute('''CREATE TABLE data (id INTEGER PRIMARY KEY AUTOINCREMENT, name TEXT, age INTEGER)''')

步骤5：插入数据

使用下面的代码段从CSV文件插入数据：

with open('data.csv', 'r') as f:
    reader = csv.reader(f)
    for row in reader:
        name, age = row
        cursor.execute('''INSERT INTO data (name, age) VALUES (?, ?)''', (name, age))

步骤6：提交更改

当完成对数据库进行更改时，请确保提交更改。

conn.commit()

步骤7：关闭连接

最后，使用以下代码段关闭数据库连接：

conn.close()

完整代码

import csv
import sqlite3

conn = sqlite3.connect('database.db')
cursor = conn.cursor()

cursor.execute('''CREATE TABLE data (id INTEGER PRIMARY KEY AUTOINCREMENT, name TEXT, age INTEGER)''')

with open('data.csv', 'r') as f:
    reader = csv.reader(f)
    for row in reader:
        name, age = row
        cursor.execute('''INSERT INTO data (name, age) VALUES (?, ?)''', (name, age))

conn.commit()
conn.close()

这样就完成了从CSV创建SQLite数据库的过程。你可以改变数据源和表结构，以符合你的项目需求。