使用Python生成新春烟花效果的方法

pip install pygame

import turtle
import random
import math
 
# 设置屏幕
screen = turtle.Screen()
screen.bgcolor("black")
screen.title("新春烟花效果")
screen.tracer(0)
 
# 创建烟花粒子
class Firework:
    def __init__(self, x, y, color):
        self.particles = []
        self.color = color
        for _ in range(100):  # 每次生成100个粒子
            angle = random.uniform(0, 2 * math.pi)
            speed = random.uniform(2, 5)
            dx = math.cos(angle) * speed
            dy = math.sin(angle) * speed
            self.particles.append([x, y, dx, dy, 1.0])  # (x, y, dx, dy, alpha)
 
    def update(self):
        new_particles = []
        for particle in self.particles:
            x, y, dx, dy, alpha = particle
            if alpha > 0:
                x += dx
                y += dy
                dy -= 0.05  # 模拟重力
                alpha -= 0.02  # 粒子逐渐消失
                new_particles.append([x, y, dx, dy, alpha])
        self.particles = new_particles
 
    def draw(self, pen):
        for x, y, _, _, alpha in self.particles:
            pen.goto(x, y)
            pen.dot(5, (self.color[0], self.color[1], self.color[2], alpha))
 
# 创建烟花管理器
class FireworkManager:
    def __init__(self):
        self.fireworks = []
 
    def add_firework(self, x, y, color):
        self.fireworks.append(Firework(x, y, color))
 
    def update(self):
        new_fireworks = []
        for firework in self.fireworks:
            firework.update()
            if firework.particles:
                new_fireworks.append(firework)
        self.fireworks = new_fireworks
 
    def draw(self, pen):
        for firework in self.fireworks:
            firework.draw(pen)
 
# 初始化
manager = FireworkManager()
pen = turtle.Turtle()
pen.hideturtle()
pen.speed(0)
 
# 主循环
def main_loop():
    screen.update()
    pen.clear()
    if random.random() < 0.1:  # 随机生成烟花
        x = random.randint(-300, 300)
        y = random.randint(100, 300)
        color = (random.random(), random.random(), random.random())
        manager.add_firework(x, y, color)
    manager.update()
    manager.draw(pen)
    screen.ontimer(main_loop, 50)
 
main_loop()
screen.mainloop()

import pygame
import random
import math
 
# 初始化
pygame.init()
screen = pygame.display.set_mode((800, 600))
pygame.display.set_caption("新春烟花效果")
clock = pygame.time.Clock()
 
# 烟花粒子类
class Particle:
    def __init__(self, x, y, color):
        self.x = x
        self.y = y
        self.color = color
        self.radius = random.randint(3, 6)
        self.speed = random.uniform(2, 6)
        self.angle = random.uniform(0, 2 * math.pi)
        self.alpha = 255
 
    def update(self):
        self.x += math.cos(self.angle) * self.speed
        self.y += math.sin(self.angle) * self.speed
        self.speed *= 0.98  # 减速效果
        self.alpha -= 4  # 逐渐消失
 
    def draw(self, surface):
        if self.alpha > 0:
            s = pygame.Surface((self.radius * 2, self.radius * 2), pygame.SRCALPHA)
            pygame.draw.circle(s, (*self.color, self.alpha), (self.radius, self.radius), self.radius)
            surface.blit(s, (self.x - self.radius, self.y - self.radius))
 
# 烟花管理器
class FireworkManager:
    def __init__(self):
        self.particles = []
 
    def add_firework(self, x, y, color):
        for _ in range(100):  # 每次生成100个粒子
            self.particles.append(Particle(x, y, color))
 
    def update(self):
        self.particles = [p for p in self.particles if p.alpha > 0]
        for p in self.particles:
            p.update()
 
    def draw(self, surface):
        for p in self.particles:
            p.draw(surface)
 
manager = FireworkManager()
 
# 主循环
running = True
while running:
    screen.fill((0, 0, 0))
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
 
    if random.random() < 0.05:
        x = random.randint(100, 700)
        y = random.randint(100, 300)
        color = (random.randint(100, 255), random.randint(100, 255), random.randint(100, 255))
        manager.add_firework(x, y, color)
 
    manager.update()
    manager.draw(screen)
 
    pygame.display.flip()
    clock.tick(30)
 
pygame.quit()