在pygame中有没有一种方法可以使用for循环来制作游戏地图？

import pygame

# 初始化Pygame
pygame.init()

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

# 定义地图大小和格子大小
map_width = 20
map_height = 20
tile_size = 40

# 创建一个二维数组来表示地图
game_map = [
    [1, 1, 1, 1, 1],
    [1, 0, 0, 0, 1],
    [1, 0, 1, 0, 1],
    [1, 0, 0, 0, 1],
    [1, 1, 1, 1, 1]
]

# 定义颜色
white = (255, 255, 255)
black = (0, 0, 0)

# 游戏主循环
running = True
while running:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False

    # 绘制地图
    for row in range(map_height):
        for col in range(map_width):
            color = white if game_map[row][col] == 0 else black
            pygame.draw.rect(screen, color, (col * tile_size, row * tile_size, tile_size, tile_size))

    # 更新屏幕
    pygame.display.flip()

# 退出Pygame
pygame.quit()