python 将csv文件写入数据库
时间: 2024-06-20 11:02:51 浏览: 255
将 CSV 文件写入数据库通常需要先将 CSV 文件读取到内存中,然后再将其写入到数据库中。以下是 Python 将 CSV 文件写入数据库的基本步骤:
1. 连接数据库,打开游标。
2. 打开 CSV 文件,读取数据。
3. 遍历 CSV 数据,将每一行数据插入到数据库中。
4. 关闭 CSV 文件和数据库连接。
下面是一个示例代码,假设你要将一个名为 `example.csv` 的文件写入名为 `example_table` 的数据库表中。
```python
import csv
import sqlite3
# 连接数据库
conn = sqlite3.connect('example.db')
cursor = conn.cursor()
# 创建表格
cursor.execute('''CREATE TABLE example_table (id INTEGER PRIMARY KEY, name TEXT, age INTEGER)''')
# 打开 CSV 文件
with open('example.csv', 'r') as f:
reader = csv.reader(f)
next(reader) # 跳过标题行
# 遍历 CSV 数据,将每一行数据插入到数据库中
for row in reader:
cursor.execute("INSERT INTO example_table (name, age) VALUES (?, ?)", (row, row))
# 提交更改并关闭连接
conn.commit()
conn.close()
```
阅读全文