帮我编写一个烟花代码编程python满屏

创建一个满屏烟花效果的Python程序通常会涉及动画和图形绘制。我们将使用Python的pygame库来实现这个效果。首先，确保已安装pygame，如果没有，可以用`pip install pygame`来安装。

以下是一个简单的烟花动画示例：

import pygame
import time

# 初始化pygame
pygame.init()

# 设置窗口尺寸
screen_width, screen_height = 800, 600
screen = pygame.display.set_mode((screen_width, screen_height))

# 烟花颜色和位置
fireworks_colors = [(255, 0, 0), (0, 255, 0), (0, 0, 255)]
firework_positions = [(x, y) for x in range(screen_width) for y in range(screen_height)]

def draw_firework(x, y, color):
    pygame.draw.circle(screen, color, (x, y), 50, 0)

while True:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            exit()

    # 清空屏幕
    screen.fill((0, 0, 0))

    # 随机选择一个烟花发射
    firework_color = fireworks_colors[pygame.time.get_ticks() % len(fireworks_colors)]
    firework_position = firework_positions.pop(0)
    
    draw_firework(*firework_position, firework_color)
    
    # 添加新的烟花
    for _ in range(2):  # 每次循环增加两个新的烟花
        random_x = screen_width // 2 + pygame.math.randint(-100, 100)
        random_y = screen_height // 2 + pygame.math.randint(-100, 100)
        firework_positions.append((random_x, random_y))
        
    # 更新屏幕显示
    pygame.display.update()
    time.sleep(0.02)  # 控制烟花绽放的速度