如何把csv导入mysql

原创

mob649e8167c4a3 2023-07-27 05:02:40 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e8167c4a3的原创作品，请联系作者获取转载授权，否则将追究法律责任

如何将CSV导入MySQL

在数据处理和分析过程中，CSV（逗号分隔值）是一种常见的数据格式。而MySQL是一个流行的关系型数据库管理系统，提供了强大的数据存储和查询功能。将CSV文件导入MySQL数据库可以方便地进行数据管理和查询。本文将介绍如何使用Python和MySQL连接器将CSV文件导入MySQL数据库。

实际问题

假设我们有一个包含学生信息的CSV文件，其中包括学生的学号、姓名、年龄和性别。我们需要将这些学生信息导入MySQL数据库中的一个表格，以便进一步分析和管理。

示例

以下是一个示例CSV文件（students.csv），包含4个字段：学号、姓名、年龄和性别。

学号,姓名,年龄,性别
1,张三,18,男
2,李四,19,女
3,王五,20,男
4,赵六,18,男
5,钱七,19,女

首先，我们需要安装Python和MySQL连接器。可以使用以下命令安装MySQL连接器：

pip install mysql-connector-python

接下来，我们需要创建一个MySQL数据库和一个表格来存储学生信息。可以使用以下SQL语句在MySQL中创建数据库和表格：

CREATE DATABASE students;
USE students;
CREATE TABLE student_info (
  id INT AUTO_INCREMENT PRIMARY KEY,
  学号 INT NOT NULL,
  姓名 VARCHAR(255) NOT NULL,
  年龄 INT NOT NULL,
  性别 VARCHAR(10) NOT NULL
);

导入CSV文件

我们将使用Python来读取CSV文件并将数据插入MySQL数据库。

首先，我们需要导入必要的库：

import csv
import mysql.connector

接下来，我们需要建立与MySQL数据库的连接：

mydb = mysql.connector.connect(
  host="localhost",
  user="yourusername",
  password="yourpassword",
  database="students"
)

然后，我们可以创建一个游标对象来执行SQL语句：

mycursor = mydb.cursor()

现在，我们可以打开CSV文件并读取数据行。使用以下代码：

with open('students.csv', 'r') as file:
    reader = csv.reader(file)
    next(reader)  # 跳过CSV文件的标题行
    for row in reader:
        学号, 姓名, 年龄, 性别 = row  # 解析CSV行中的数据
        sql = "INSERT INTO student_info (学号, 姓名, 年龄, 性别) VALUES (%s, %s, %s, %s)"
        val = (学号, 姓名, 年龄, 性别)
        mycursor.execute(sql, val)  # 执行插入语句
        mydb.commit()  # 提交更改

以上代码将逐行读取CSV文件，并将每行数据插入到MySQL数据库的student_info表格中。

最后，我们需要关闭数据库连接：