python代码烟花

Python代码可以用来制作炫酷的烟花秀。当按下键1时，会立即生成一个新的“烟花”；当按下键2时，会同时生成10个“烟花”。下面是示例代码：

import pygame
from firework import Firework

pygame.init()

# Set up the display
screen = pygame.display.set_mode((800, 600))
pygame.display.set_caption("Fireworks")

# Set up the clock
clock = pygame.time.Clock()

# Set up the list of fireworks
fireworks = []

# Main game loop
running = True
while running:
    # Handle events
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
        if event.type == pygame.KEYDOWN:
            # Change game speed with number keys
            if event.key == pygame.K_1:
                fireworks.append(Firework())
            if event.key == pygame.K_2:
                for i in range(10):
                    fireworks.append(Firework())

    # Update fireworks
    for firework in fireworks:
        firework.update()

    # Draw fireworks
    screen.fill((0, 0, 0))
    for firework in fireworks:
        firework.draw(screen)

    # Update the display
    pygame.display.flip()

    # Tick the clock
    clock.tick(60)

# Clean up
pygame.quit()