我用Python做了个植物大战僵尸(附代码)

import pygame  import random  import sys  import time  # 初始化pygame  pygame.init()  # 设置颜色  WHITE = (255, 255, 255)  GREEN = (0, 255, 0)  RED = (255, 0, 0)  # 设置屏幕大小  SCREEN_WIDTH, SCREEN_HEIGHT = 800, 600screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))  # 植物和僵尸的类  classPlant:def__init__(self, x, y):        self.x = x          self.y = y          self.bullets = []  defshoot(self):        bullet = pygame.Rect(self.x + 20, self.y, 5, 10)          self.bullets.append(bullet)  defmove_bullets(self):for bullet in self.bullets:              bullet.y -= 5defdraw(self):        pygame.draw.rect(screen, GREEN, pygame.Rect(self.x, self.y, 50, 50))  for bullet in self.bullets:              pygame.draw.rect(screen, RED, bullet)  classZombie:def__init__(self):        self.x = random.randint(SCREEN_WIDTH, SCREEN_WIDTH + 50)          self.y = random.randint(0, SCREEN_HEIGHT)          self.speed = random.randint(1, 5)  defmove(self):        self.x -= self.speed  defdraw(self):        pygame.draw.rect(screen, RED, pygame.Rect(self.x, self.y, 50, 50))  # 游戏主循环  defgame_loop():    plant = Plant(SCREEN_WIDTH - 60, SCREEN_HEIGHT // 2)      zombies = [Zombie() for _ in range(5)]      running = Truewhile running: