python sqlite 导出csv

python sqlite 导出csv

在数据分析和处理过程中，有时候我们需要从数据库中提取数据并将其导出为CSV文件，以便于后续的处理和分析。在Python中，我们可以使用SQLite库来连接数据库，并通过执行SQL查询来获取数据，然后将数据导出为CSV文件。

连接SQLite数据库

首先，我们需要安装SQLite库，可以使用以下命令来安装：

pip install sqlite3

接下来，我们可以通过以下代码来连接SQLite数据库：

import sqlite3

# 连接数据库
connection = sqlite3.connect('example.db')
cursor = connection.cursor()

# 创建表
cursor.execute('''CREATE TABLE IF NOT EXISTS users 
                  (id integer PRIMARY KEY, name text, age integer)''')

# 插入数据
cursor.execute("INSERT INTO users (name, age) VALUES ('Alice', 30)")
cursor.execute("INSERT INTO users (name, age) VALUES ('Bob', 25)")

# 提交更改并关闭连接
connection.commit()
connection.close()

以上代码创建了一个名为example.db的SQLite数据库，并在其中创建了一个名为users的表，然后插入了两条数据。

导出数据为CSV文件

接下来，我们可以编写代码来查询数据并将其导出为CSV文件：

import sqlite3
import csv

# 连接数据库
connection = sqlite3.connect('example.db')
cursor = connection.cursor()

# 执行查询
cursor.execute("SELECT * FROM users")
data = cursor.fetchall()

# 将数据导出到CSV文件
with open('users.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    # 写入表头
    writer.writerow(['id', 'name', 'age'])
    # 写入数据
    writer.writerows(data)

# 关闭连接
connection.close()

以上代码执行了一个查询，将users表中的所有数据取出，并将数据导出为users.csv文件。在导出数据时，我们先写入表头，然后写入数据行。

运行结果

运行以上代码后，会生成一个名为users.csv的CSV文件，其内容如下：

id,name,age
1,Alice,30
2,Bob,25

这就是将SQLite数据库中的数据导出为CSV文件的方法，通过这种方式我们可以方便地将数据用于其他分析工具或平台中。