如何在Python中生成烟花代码的动画效果？

在Python中生成烟花代码的动画效果，通常会使用一些图形库，比如pygame或者matplotlib的animation模块。以下是使用pygame的一个简单示例：

import pygame
import random

# 初始化pygame
pygame.init()

# 设置窗口大小和标题
screen = pygame.display.set_mode((640, 480))
pygame.display.set_caption("Pygame Fireworks")

# 定义颜色
colors = [(255, 0, 0), (0, 255, 0), (0, 0, 255)]

def draw_firework(x, y):
    # 创建随机大小的圆形
    size = random.randint(10, 30)
    pygame.draw.circle(screen, random.choice(colors), (x, y), size)

# 主循环
while True:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            quit()

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

    # 随机生成烟花位置和绘制
    fireworks = [(random.randint(0, 640), random.randint(0, 480)) for _ in range(5)]
    for firework in fireworks:
        draw_firework(*firework)

    # 更新显示
    pygame.display.flip()

这个例子中，程序会在窗口上随机生成并绘制多个彩色火花，每次循环更新一次屏幕。你可以根据需求调整颜色选择、烟花的数量和样式。