如何用Python将许多csv文件存入sqlite3
时间: 2023-06-04 16:06:56 浏览: 267
python解决解决_sqlite3资源复制到对应文件夹中即可
数据库?
您可以使用Python中的csv和sqlite3模块来完成这个任务。以下是一个示例代码:
```python
import csv
import sqlite3
import os
# 打开数据库连接
conn = sqlite3.connect('database.db')
# 创建游标对象
cursor = conn.cursor()
# 创建表格
cursor.execute('''CREATE TABLE IF NOT EXISTS data
(ID INTEGER PRIMARY KEY,
Col1 TEXT,
Col2 TEXT,
Col3 TEXT);''')
# 读取csv文件并将数据存入数据库
path = '/path/to/csv/files/'
for filename in os.listdir(path):
if filename.endswith('.csv'):
with open(os.path.join(path, filename), 'r') as csvfile:
reader = csv.reader(csvfile)
next(reader) # 跳过首行
for row in reader:
cursor.execute("INSERT INTO data (Col1, Col2, Col3) VALUES (?, ?, ?)", (row[0], row[1], row[2]))
# 提交到数据库
conn.commit()
# 关闭游标和连接
cursor.close()
conn.close()
```
如果您有多个csv文件需要存入数据库,只需将它们存放在同一个文件夹中,并调整 path 变量的值。此代码将跳过每个文件的首行(假设它们是标题行),并将数据存入名为 data 的表格中,其中包含三个列。您可以根据自己的需要修改表格和列的名称。
阅读全文