怎么将csv表格导入到sqlite数据库里|极客教程

怎么将csv表格导入到sqlite数据库里

在数据分析和数据处理中，经常需要将从不同来源获取的数据整合到一起进行处理。CSV(Comma Separated Values)是一种常见的数据存储格式，而SQLite是一种轻量级的关系型数据库管理系统。在本文中，我们将介绍如何将CSV表格导入到SQLite数据库里，从而方便进行数据分析和处理。

准备工作

在开始之前，我们需要确保已经安装好了SQLite数据库和相关的Python库。可以通过以下命令安装sqlite3和pandas：

pip install pandas

导入CSV表格到SQLite数据库

在导入CSV表格到SQLite数据库之前，需要创建一个数据库和数据表来存储数据。我们可以使用sqlite3模块来操作SQLite数据库，pandas库可以用来读取CSV文件，并提供了直接将数据导入数据库的方法。

以下是一个完整的示例代码，演示了如何将一个名为data.csv的CSV表格导入到一个名为mydatabase.db的SQLite数据库里：

import sqlite3
import pandas as pd

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

# 创建一个名为data的表格
c.execute('''CREATE TABLE data
             ([Date] DATE, [Sales] FLOAT)''')

# 读取CSV文件并将数据插入到表格里
df = pd.read_csv('data.csv')
df.to_sql('data', conn, if_exists='replace', index = False)

# 提交并关闭数据库连接
conn.commit()
conn.close()

在上面的代码中，首先通过sqlite3.connect()方法连接到SQLite数据库，并使用cursor()方法创建一个游标对象。然后使用execute()方法创建一个名为data的表格，表格包含两列：Date和Sales。接着使用pd.read_csv()方法读取CSV文件，然后调用to_sql()方法将数据插入到数据库里。最后使用commit()方法提交更改，并关闭数据库连接。

查询数据

一旦数据已经导入到SQLite数据库里，我们就可以通过执行SQL查询语句来查看数据。以下是一个示例代码，演示了如何查询数据库里的所有数据：

import sqlite3

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

# 查询data表格里的所有数据
c.execute('SELECT * FROM data')
rows = c.fetchall()

# 输出查询结果
for row in rows:
    print(row)

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

在上面的代码中，首先通过sqlite3.connect()方法连接到数据库，并创建一个游标对象。然后执行SELECT * FROM data查询语句，将查询结果赋值给rows变量。最后遍历rows变量并打印每一行数据，最后调用close()方法关闭数据库连接。