在python中设置记录集的批次号

import pymysql

# 连接到数据库
connection = pymysql.connect(host='localhost',
                             user='username',
                             password='password',
                             db='database_name')

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

    # 执行查询操作
    query = "SELECT * FROM table_name"
    cursor.execute(query)

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

    # 为每个记录设置批次号
    batch_number = 1
    for row in result:
        # 更新记录的批次号字段
        update_query = "UPDATE table_name SET batch_number=%s WHERE id=%s"
        cursor.execute(update_query, (batch_number, row['id']))
        batch_number += 1

    # 提交修改
    connection.commit()

finally:
    # 关闭连接
    connection.close()