用phthon写一个读取文件夹内CSV文件数据到数据库的代码
时间: 2024-03-26 12:38:28 浏览: 75
好的,以下是一个示例代码,可以将指定文件夹内的CSV文件数据读取到数据库中:
```python
import os
import csv
import sqlite3
# 连接数据库
conn = sqlite3.connect('test.db')
# 创建数据表
conn.execute('''CREATE TABLE IF NOT EXISTS csv_data
(ID INTEGER PRIMARY KEY AUTOINCREMENT,
NAME TEXT,
AGE INT,
GENDER TEXT);''')
# 指定CSV文件所在的文件夹路径
folder_path = 'path/to/csv/folder'
# 遍历文件夹,读取CSV文件数据并插入数据库
for file_name in os.listdir(folder_path):
if file_name.endswith('.csv'):
with open(os.path.join(folder_path, file_name), 'r') as f:
reader = csv.reader(f)
next(reader) # 跳过CSV文件的表头行
for row in reader:
name, age, gender = row
conn.execute(f"INSERT INTO csv_data (NAME, AGE, GENDER) VALUES ('{name}', {age}, '{gender}');")
# 提交数据库操作并关闭连接
conn.commit()
conn.close()
```
需要注意的是,这只是一个简单的示例代码,实际应用中还需要进行异常处理、SQL注入防范等操作。同时,如果CSV文件中的数据类型不确定,需要进行类型转换和数据清洗等操作。
阅读全文