用CSV文件快速转入MySQL数据库(csv文件转入mysql)
用CSV文件快速转入MySQL数据库
在数据分析和处理中,常常需要将大量数据从CSV文件导入到MySQL数据库中进行处理和存储。本文将介绍如何使用Python编写脚本来快速地将CSV文件转移至MySQL数据库中。
步骤1: 安装MySQL驱动模块
在Python环境中安装MySQL驱动模块pymysql,这里我们使用pip install pymysql命令进行安装。
步骤2: 创建数据库和表结构
在MySQL中创建一个数据库,以及用于存储CSV数据的表。下面是创建表的具体代码。
import pymysql
db = pymysql.connect(“localhost”,”root”,”root123″,”testdb” )
cursor = db.cursor()
sql = “””CREATE TABLE EMPLOYEE (
FIRST_NAME CHAR(20) NOT NULL,
LAST_NAME CHAR(20),
AGE INT,
SEX CHAR(1),
INCOME FLOAT )”””
cursor.execute(sql)
db.close()
步骤3: 读取CSV文件中的数据
接下来,我们需要读取CSV文件中的数据。Python内置了csv模块来实现CSV文件的读写操作。我们利用csv模块读取CSV文件内容,下面是具体代码。
import csv
with open(’employee.csv’, newline=”) as csvfile:
reader = csv.DictReader(csvfile)
for row in reader:
print(row)
步骤4: 将数据插入到MySQL表中
有了CSV文件的内容,我们就可以将数据插入到MySQL的表中。下面是具体代码。
import pymysql
import csv
db = pymysql.connect(“localhost”,”root”,”root123″,”testdb” )
cursor = db.cursor()
sql = “””INSERT INTO EMPLOYEE(FIRST_NAME,
LAST_NAME, AGE, SEX, INCOME)
VALUES (%s, %s, %s, %s, %s)”””
with open(’employee.csv’, newline=”) as csvfile:
reader = csv.DictReader(csvfile)
for row in reader:
cursor.execute(sql, (row[‘FIRST_NAME’],
row[‘LAST_NAME’],
row[‘AGE’],
row[‘SEX’],
row[‘INCOME’]))
db.commit()
db.close()
通过以上几个步骤,我们就可以将CSV文件中的数据快速载入到MySQL数据库中,实现对数据的处理和存储。使用Python脚本进行操作,不仅可以提高工作效率,而且可以方便的重复执行使得数据的导入变得更加快捷。