用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脚本进行操作,不仅可以提高工作效率,而且可以方便的重复执行使得数据的导入变得更加快捷。


数据运维技术 » 用CSV文件快速转入MySQL数据库(csv文件转入mysql)