Python 游标

Python 中的游标(cursor)是一种用来从数据库中检索和操作数据的对象，它允许我们在执行 SQL 查询时，控制数据的获取和处理方式。

一、游标的基本概念

游标通常与数据库连接一起使用，用于执行查询、获取结果集或者更新数据。在 Python 中，我们可以使用各种数据库驱动程序来创建和使用游标，例如使用 `pymysql` 模块来连接 MySQL 数据库。

<!-- 安装 pymysql 模块 -->
$ pip install pymysql

首先，我们需要创建一个数据库连接：

import pymysql

# 创建连接
conn = pymysql.connect(host='localhost', port=3306, user='root', passwd='password', db='mydb')

# 创建游标
cursor = conn.cursor()

创建游标之后，我们可以执行 SQL 查询并获取结果集：

# 执行查询
cursor.execute("SELECT * FROM students")

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

# 处理结果集
for row in results:
    print(row)

当我们使用游标执行查询时，可以通过调用 `fetchall()` 方法来获取全部结果，或者通过调用 `fetchone()` 方法来逐个获取结果。

二、游标的使用技巧

1、查询结果分页

在处理大量数据时，通常需要将结果按页展示。我们可以利用游标的 `fetchmany()` 方法来实现查询结果的分页显示。

# 执行查询
cursor.execute("SELECT * FROM students")

# 获取第一页结果
results = cursor.fetchmany(10)

# 处理结果
for row in results:
    print(row)

2、更新数据

游标不仅可以用于查询数据，还可以用于更新、删除和插入数据。我们可以使用游标的 `execute()` 方法来执行更新操作。

# 执行更新操作
cursor.execute("UPDATE students SET age = 18 WHERE name = 'Tom'")

# 提交事务
conn.commit()

3、事务处理

在对数据库进行操作时，有些时候需要保证一系列操作的原子性。游标提供了在事务中执行多个操作的功能。

# 开始事务
conn.begin()

try:
    # 执行操作1
    cursor.execute("INSERT INTO students (name, age) VALUES ('Alice', 20)")
    
    # 执行操作2
    cursor.execute("DELETE FROM students WHERE age < 18")
    
    # 提交事务
    conn.commit()
except Exception as e:
    # 回滚事务
    conn.rollback()

三、总结

本文介绍了 Python 中的游标(cursor)的基本概念和使用技巧。通过使用游标，我们可以控制数据查询和操作的方式，并且能够实现结果分页、数据更新和事务处理等功能。希望本文对你理解和使用 Python 游标有所帮助。