MySQL中使用Python导入CSV数据|极客教程

MySQL中使用Python导入CSV数据

在MySQL数据库中，可以使用LOAD DATA命令导入CSV文件中的数据。但是，这种方法需要一定的SQL编程知识。因此，本文将介绍如何使用Python来导入CSV数据到MySQL数据库中。

阅读更多：MySQL 教程

准备工作

在开始本教程之前，我们需要做一些准备工作。

安装MySQL Connector Python

MySQL Connector Python是Python与MySQL数据库之间的一个标准接口，用于连接和操作MySQL数据库。需要先安装这个模块才能在Python中连接MySQL数据库。

可以在Python官方网站上下载并安装MySQL Connector Python：https://dev.mysql.com/downloads/connector/python/

准备CSV文件

我们需要一个包含要导入MySQL数据库中的数据的CSV文件。

例如，在以下路径中有一个包含学生信息的CSV文件：

/home/user/students.csv

该文件的内容如下：

id,name,age,email
1,Tom,18,tom@email.com
2,Jack,19,jack@email.com
3,Sam,20,sam@email.com

使用Python导入CSV数据

在安装好MySQL Connector Python和准备好CSV文件之后，我们可以编写Python脚本来导入CSV数据到MySQL数据库中。

以下示例代码展示了如何使用Python导入CSV数据到MySQL数据库中：

import os
import csv
import mysql.connector

def import_data_from_csv_to_mysql():
    # 设置MySQL连接参数
    mysql_host = "localhost"
    mysql_user = "root"
    mysql_password = "password"
    mysql_database = "mydatabase"

    # 连接MySQL数据库
    conn = mysql.connector.connect(host=mysql_host, user=mysql_user, password=mysql_password, database=mysql_database)

    # 获取MySQL游标
    cursor = conn.cursor()

    # 准备SQL语句
    sql = "INSERT INTO students (id, name, age, email) VALUES (%s, %s, %s, %s)"

    # 打开CSV文件并读取数据
    with open('/home/user/students.csv') as csv_file:
        csv_reader = csv.reader(csv_file, delimiter=',')
        next(csv_reader)
        for row in csv_reader:
            # 插入数据到MySQL数据库
            cursor.execute(sql, tuple(row))

    # 提交数据到MySQL数据库
    conn.commit()

    # 关闭连接和游标
    cursor.close()
    conn.close()

if __name__ == '__main__':
    import_data_from_csv_to_mysql()
    print("CSV数据导入MySQL数据库成功！")