python csv 存入mysql
时间: 2023-05-04 09:03:57 浏览: 139
在使用Python存储CSV文件到MySQL之前,我们需要安装pymysql或者MySQL驱动程序,这些库可以使用“pip install pymysql”或“pip install mysql-connector-python”安装。 接下来,我们需要连接到MySQL数据库并创建一个表来存储CSV数据。 我们可以使用以下Python代码来连接并创建一个名为“mytable”的表:
```
import pymysql
# 连接到数据库
conn = pymysql.connect(host='localhost', user='root', password='password', db='mydatabase')
# 获取游标
cursor = conn.cursor()
# 创建表
stmt = "CREATE TABLE mytable (id INT PRIMARY KEY, name VARCHAR(255), email VARCHAR(255))"
cursor.execute(stmt)
# 关闭游标和连接
cursor.close()
conn.close()
```
接下来,我们需要将CSV文件中的数据导入表中。我们可以读取CSV文件,当每一行的数据插入到MySQL表中。以下是代码示例:
```
import csv
import pymysql
# 连接到数据库
conn = pymysql.connect(host='localhost', user='root', password='password', db='mydatabase')
# 获取游标
cursor = conn.cursor()
# 打开CSV文件
with open('data.csv', 'r') as csvfile:
reader = csv.reader(csvfile)
next(reader) # 忽略 CSV 文件的标题行
for row in reader:
# 插入数据到 MySQL 表
stmt = "INSERT INTO mytable (id, name, email) VALUES (%s, %s, %s)"
cursor.execute(stmt, (int(row[0]), row[1], row[2]))
# 提交到数据库并关闭连接
conn.commit()
cursor.close()
conn.close()
```
这个例子中,我们使用了Python的csv库来读取CSV文件中的数据,并使用INSERT语句将数据插入到已经创建的MySQL表中。最后,我们提交更改并关闭与MySQL数据库的连接。
阅读全文