SQLite 使用Python从sqlite3数据库中写入CSV

在本文中，我们将介绍如何使用Python从SQLite3数据库中将数据写入CSV文件。

阅读更多：SQLite 教程

1. 连接到SQLite数据库

首先，我们需要在Python中连接到SQLite数据库。我们可以使用sqlite3模块来实现这一点。下面是一个连接到数据库的示例代码：

import sqlite3

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

在这里，我们使用sqlite3.connect()函数连接到名为database.db的数据库。如果数据库文件不存在，它将被创建。

2. 执行SQL查询

接下来，我们需要执行一个SQL查询来检索要写入CSV文件的数据。在这里，我们将使用SQLite的SELECT语句。下面是一个示例代码：

cursor = conn.cursor()

query = 'SELECT * FROM table_name'
cursor.execute(query)

rows = cursor.fetchall()

在这里，我们使用cursor()方法创建一个游标对象来执行SQL查询。然后，我们使用cursor.execute()方法来执行SELECT语句，并使用cursor.fetchall()方法获取所有结果行。

3. 写入CSV文件

一旦我们获得了要写入CSV文件的数据，我们可以使用Python的csv模块来实现写入操作。下面是一个示例代码：

import csv

filename = 'output.csv'

with open(filename, 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow([i[0] for i in cursor.description])
    writer.writerows(rows)

在这里，我们使用csv.writer()方法创建一个写入器对象来写入CSV文件。然后，我们使用writer.writerow()方法写入查询结果的列名。最后，我们使用writer.writerows()方法写入所有结果行。

4. 完整示例

下面是一个完整的示例代码，将SQLite数据库中的数据写入CSV文件：

import sqlite3
import csv

# 连接到SQLite数据库
conn = sqlite3.connect('database.db')

# 执行SQL查询
cursor = conn.cursor()
query = 'SELECT * FROM table_name'
cursor.execute(query)
rows = cursor.fetchall()

# 写入CSV文件
filename = 'output.csv'
with open(filename, 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow([i[0] for i in cursor.description])
    writer.writerows(rows)

# 关闭数据库连接
conn.close()

总结

在本文中，我们介绍了如何使用Python从SQLite3数据库中将数据写入CSV文件。我们使用sqlite3模块连接到数据库，执行SQL查询并获取结果。然后，我们使用csv模块将结果写入CSV文件。这个过程简单而有效，为我们提供了一种方便的方法来导出SQLite数据库中的数据。希望本文对你有所帮助！