pygame将当前画面输出到变量

from PIL import Image
import pygame
import numpy as np
 
pygame.display.update() # 更新画面
pil_string_image = pygame.image.tostring(screen, "RGB", False)  # 将当前图像输出为字符变量
pli_image = Image.frombytes('RGB', screen.get_size(), pil_string_image, 'raw') # 将字符变量变为Image变量
img = np.array(pli_image) # 转变为numpy
small = pli_image.resize((100, 100), resample=Image.BOX) # 改变尺寸
small.show() 
# 保存本地看一下
pli_image.save('{}.jpg'.format(index))