Python 量化数据仓库搭建系列 2：操作 MySQL 数据库

前言

上一篇文章中，我们介绍了如何使用 Python 连接到 MySQL 数据库。本篇文章将继续讲解如何在 Python 中操作 MySQL 数据库，包括查询、增删改查等基本操作。此外，我们还将封装一些常见的方法，以便您能够更轻松地与 MySQL 数据库进行交互。

准备工作

首先，您需要确保已经安装了 Python 和 MySQLdb 模块。如果尚未安装，请使用以下命令进行安装：

pip install mysql-connector-python

此外，您还需要创建一个 MySQL 数据库并创建一个用户，以便 Python 脚本可以连接到数据库。

连接到数据库

import mysql.connector

# 创建连接对象
connection = mysql.connector.connect(
    host="localhost",
    user="username",
    password="password",
    database="database_name"
)

# 创建游标对象
cursor = connection.cursor()

查询数据

# 查询所有数据
cursor.execute("SELECT * FROM table_name")

# 获取查询结果
results = cursor.fetchall()

# 打印查询结果
for row in results:
    print(row)

插入数据

# 插入一条数据
cursor.execute("INSERT INTO table_name (column1, column2, column3) VALUES (%s, %s, %s)", (value1, value2, value3))

# 提交事务
connection.commit()

更新数据

# 更新一条数据
cursor.execute("UPDATE table_name SET column1 = %s, column2 = %s WHERE id = %s", (value1, value2, id))

# 提交事务
connection.commit()

删除数据

# 删除一条数据
cursor.execute("DELETE FROM table_name WHERE id = %s", (id,))

# 提交事务
connection.commit()

封装方法

为了更方便地操作 MySQL 数据库，我们可以封装一些常见的方法。例如，我们可以封装一个查询方法，如下所示：

def query(sql, params=None):
    """
    查询数据

    Args:
        sql: SQL 查询语句
        params: 查询参数

    Returns:
        查询结果
    """

    # 创建游标对象
    cursor = connection.cursor()

    # 执行查询
    cursor.execute(sql, params)

    # 获取查询结果
    results = cursor.fetchall()

    # 关闭游标对象
    cursor.close()

    return results

使用这个封装的方法，我们可以更轻松地查询数据。例如，我们可以使用以下代码查询所有用户：

users = query("SELECT * FROM users")

for user in users:
    print(user)

结语

本篇文章介绍了如何使用 Python 操作 MySQL 数据库，包括查询、增删改查等基本操作。此外，我们还封装了一些常见的方法，以便您能够更轻松地与 MySQL 数据库进行交互。希望本篇文章能够对您有所帮助。