使用 Python 和 PyMySQL 遍历查询结果集

Python 是一种广泛使用的高级编程语言，具有易于学习和使用的特点。PyMySQL 是一个用于与 MySQL 数据库进行交互的库，它允许我们通过 Python 执行 SQL 查询、操作数据库、获取数据等。在本篇文章中，我们将介绍如何使用 PyMySQL 遍历查询结果集，并提供相应的代码示例。

安装 PyMySQL

在开始之前，确保你已经安装了 PyMySQL。可以通过下面的命令安装：

pip install pymysql

创建数据库和表

在进行遍历查询之前，需要有一个 MySQL 数据库和表。以下是一个创建数据库和表的 SQL 示例：

CREATE DATABASE sample_db;

USE sample_db;

CREATE TABLE employees (
    id INT AUTO_INCREMENT PRIMARY KEY,
    name VARCHAR(100),
    age INT,
    department VARCHAR(50)
);

我们插入一些示例数据：

INSERT INTO employees (name, age, department) VALUES
('Alice', 30, 'HR'),
('Bob', 25, 'Engineering'),
('Charlie', 35, 'Sales'),
('David', 28, 'Engineering'),
('Eve', 22, 'Marketing');

使用 PyMySQL 连接和查询数据库

下面的 Python 代码演示了如何使用 PyMySQL 连接到 MySQL 数据库、执行查询并遍历结果集。

import pymysql

# 连接到数据库
connection = pymysql.connect(
    host='localhost',
    user='your_username',
    password='your_password',
    database='sample_db'
)

try:
    with connection.cursor() as cursor:
        # 执行查询
        sql = "SELECT * FROM employees"
        cursor.execute(sql)

        # 获取所有结果
        results = cursor.fetchall()

        # 遍历结果集
        for row in results:
            print(f"ID: {row[0]}, Name: {row[1]}, Age: {row[2]}, Department: {row[3]}")
finally:
    connection.close()

在上面的代码中，我们首先连接到数据库，然后使用 SELECT 语句查询 employees 表中的所有数据。使用 fetchall() 方法获取所有查询结果，并通过一个 for 循环遍历每一行数据。

状态图

在执行查询后，我们可以构造一个状态图来描述程序的执行流程。下面是使用 Mermaid 语法绘制的状态图：

stateDiagram
    [*] --> ConnectDB
    ConnectDB --> ExecuteQuery
    ExecuteQuery --> FetchResults
    FetchResults --> IterateResults
    IterateResults --> [*]

这个状态图展示了从连接数据库到遍历结果的整个过程。

数据统计与可视化

在某些情况下，我们可能需要分析查询结果，并绘制相应的图表。例如，我们可以统计各个部门的员工人数，并使用饼图表示。

以下是计算各个部门员工人数并绘制饼图的代码示例：

import matplotlib.pyplot as plt
from collections import Counter

# 假设我们已经从数据库中获取所有员工数据
departments = [row[3] for row in results]
department_count = Counter(departments)

# 绘制饼图
labels = department_count.keys()
sizes = department_count.values()

plt.pie(sizes, labels=labels, autopct='%1.1f%%')
plt.title("Department Distribution")
plt.show()

小结

在本文中，我们介绍了如何使用 Python 的 PyMySQL 库连接到 MySQL 数据库、执行查询以及遍历结果集。通过实例代码，我们学习了如何从数据库中提取信息并将其输出到控制台之中。此外，我们还探讨了如何通过饼图对数据进行可视化分析。

数据库操作是编程中的重要组成部分，掌握这项技术将极大地增强你的编程能力和数据处理能力。希望本文能够为你提供帮助，让你在使用 Python 和 MySQL 时更加得心应手。