pymysql load详解|极客教程

pymysql load详解

在Python中，我们经常需要与数据库进行交互来存储和管理数据。而 pymysql 是一个用于连接和操作MySQL数据库的Python库。在本文中，我们将详细介绍如何使用 pymysql 库来实现数据的加载操作。

什么是数据加载

数据加载是指将数据从外部数据源导入到数据库中的过程。在实际应用中，我们通常会将数据从文件或其他数据库中加载到目标数据库中，以完成数据迁移、数据集成等任务。

使用pymysql加载数据的步骤

要使用 pymysql 加载数据到MySQL数据库中，通常需要以下步骤：

1. 连接数据库

首先，我们需要连接到MySQL数据库。可以使用 pymysql.connect() 方法来建立与数据库的连接。需要提供数据库的主机地址、用户名、密码、数据库名称等连接参数。

import pymysql

db = pymysql.connect(host='localhost', user='username', password='password', database='mydatabase')
cursor = db.cursor()

2. 准备数据

接下来，我们需要准备要加载的数据。数据可以是从文件中读取得到的，也可以是从其他数据库查询得到的。在本例中，我们将从文件中读取数据，并以列表的形式存储。

data = []
with open('data.txt', 'r') as file:
    for line in file:
        data.append(tuple(line.strip().split(',')))

3. 执行加载数据操作

一旦数据准备就绪，我们就可以执行加载数据的操作。使用 cursor.executemany() 方法来执行插入语句，将数据批量加载到数据库中。

sql = "INSERT INTO mytable (column1, column2) VALUES (%s, %s)"
cursor.executemany(sql, data)
db.commit()

4. 关闭数据库连接

最后，记得关闭与数据库的连接，释放资源。

db.close()

示例

假设我们有一个名为 mytable 的表，包含两列 name 和 age，我们现在有一个文件 data.txt，内容如下：

Alice,25
Bob,30
Cathy,28

我们的目标是将这些数据加载到 mytable 表中。我们可以按照上述步骤使用 pymysql 来实现这个目标。

import pymysql

db = pymysql.connect(host='localhost', user='root', password='123456', database='mydatabase')
cursor = db.cursor()

data = []
with open('data.txt', 'r') as file:
    for line in file:
        data.append(tuple(line.strip().split(',')))

sql = "INSERT INTO mytable (name, age) VALUES (%s, %s)"
cursor.executemany(sql, data)
db.commit()

db.close()

运行以上代码后，数据就会成功加载到数据库中。