Python pygame在排序函数后不会重新绘制

import pygame

# 初始化pygame
pygame.init()

# 设置屏幕大小
screen_width = 800
screen_height = 600
screen = pygame.display.set_mode((screen_width, screen_height))

# 定义排序函数
def sort_function(arr):
    # 排序算法代码
    ...

    # 在每个循环迭代中重新绘制屏幕
    for i in range(len(arr)):
        # 更新屏幕的显示
        pygame.display.flip()

        # 绘制排序后的数据
        ...

# 游戏主循环
def game_loop():
    # 初始化数据
    data = [...]

    running = True
    while running:
        # 处理事件
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                running = False

        # 在此处调用排序函数
        sort_function(data)

    # 退出游戏
    pygame.quit()

# 运行游戏循环
game_loop()