pygame simulaton另存为mp4文件

import pygame
import cv2
import numpy as np
import os

pygame.init()
width, height = 800, 600
screen = pygame.display.set_mode((width, height))

fourcc = cv2.VideoWriter_fourcc(*'mp4v')
fps = 30
video_filename = 'simulation.mp4'
video_out = cv2.VideoWriter(video_filename, fourcc, fps, (width, height))

running = True
while running:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
    
    # 游戏逻辑和绘制代码
    # ...
    
    # 将游戏场景绘制到屏幕上
    pygame.display.flip()
    
    # 将当前帧保存到视频编码器中
    screen_surface = pygame.surfarray.array3d(screen)
    screen_surface = np.transpose(screen_surface, (1, 0, 2))
    video_out.write(screen_surface)

pygame.quit()
video_out.release()