使用pymysql导入csv
时间: 2023-06-28 07:08:33 浏览: 43
可以使用以下步骤将 CSV 文件中的数据导入到 pymysql 数据库中:
1. 导入 pymysql 和 csv 模块:
```python
import pymysql
import csv
```
2. 连接数据库:
```python
conn = pymysql.connect(host='localhost', user='root', password='password', db='database_name')
```
3. 获取游标:
```python
cursor = conn.cursor()
```
4. 打开 CSV 文件:
```python
with open('data.csv', 'r') as file:
reader = csv.reader(file)
```
5. 遍历 CSV 文件中的每一行,将其插入到数据库中:
```python
for row in reader:
cursor.execute("INSERT INTO table_name (column1, column2, column3) VALUES (%s, %s, %s)", row)
```
6. 提交事务:
```python
conn.commit()
```
7. 关闭游标和连接:
```python
cursor.close()
conn.close()
```
完整代码示例:
```python
import pymysql
import csv
conn = pymysql.connect(host='localhost', user='root', password='password', db='database_name')
cursor = conn.cursor()
with open('data.csv', 'r') as file:
reader = csv.reader(file)
for row in reader:
cursor.execute("INSERT INTO table_name (column1, column2, column3) VALUES (%s, %s, %s)", row)
conn.commit()
cursor.close()
conn.close()
```