帮我写一个python烟花代码

在Python中编写一个简单的烟花动画代码可以使用pygame库来实现，因为pygame提供了丰富的图形处理功能。以下是一个基础的示例，展示如何创建烟花效果：

import pygame
import time

# 初始化pygame
pygame.init()

# 设置窗口大小
screen = pygame.display.set_mode((640, 480))
pygame.display.set_caption("简单烟花")

# 烟花的颜色和位置列表
colors = [(255, 0, 0), (0, 255, 0), (0, 0, 255)]
positions = [(100, 100), (300, 300), (500, 500)]

def draw_firework(x, y):
    size = [50, 70]
    for i in range(3):  # 颜色循环
        pygame.draw.circle(screen, colors[i], (x + size[0] // 2, y + size[1] // 2), size[i])
    pygame.display.flip()
    time.sleep(0.5)  # 暂停一段时间模拟烟花绽放过程

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

    # 更新屏幕，显示新的烟花
    for pos in positions:
        draw_firework(*pos)
        positions.remove(pos)  # 移除已经燃尽的位置

    # 如果所有位置都燃尽了，结束程序
    if not positions:
        break

pygame.quit()

运行这个代码，你会看到三个不同颜色的烟花在屏幕上依次绽放并逐渐消失。