用python软件画烟花代码

在Python中，你可以使用pygame库来创建一个简单的烟花动画效果。以下是一个基本的示例，展示如何使用pygame绘制烟花轨迹：

import pygame
import random

# 初始化pygame
pygame.init()

# 设置窗口大小
win_size = (800, 600)
screen = pygame.display.set_mode(win_size)

# 烟花颜色和速度设置
colors = [(255, 0, 0), (0, 255, 0), (0, 0, 255)]
speeds = [3, 4, 5]

# 主循环
while True:
    # 清空屏幕
    screen.fill((0, 0, 0))

    # 随机生成烟花
    for i in range(10):
        size = random.randint(10, 50)
        x, y = random.randint(0, win_size[0]), random.randint(0, win_size[1])
        direction = random.choice([-1, 1]) * speeds[random.randint(0, len(speeds) - 1)]
        color = colors[random.randint(0, len(colors) - 1)]

        # 绘制烟花轨迹
        pygame.draw.circle(screen, color, (x, y), size)
        x += direction

    # 更新屏幕
    pygame.display.update()

    # 检查是否退出游戏
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            quit()

# 结束游戏
pygame.quit()

这个代码会创建一个不断有新烟花落下的效果。注意这只是一个基础版本，你还可以添加更多的细节，比如随机颜色、轨迹变化等，来让烟花更生动。