当前位置：首页 > article >正文

基于Python实现一个浪漫烟花秀

article 2024/11/13 15:44:25

为了实现一个类似烟花秀的效果，我们可以通过复杂的粒子系统来模拟烟花的升起、绽放和下落效果。以下是一个示例，旨在创建更为动态和逼真的烟花秀效果。

示例代码

这个代码示例将使用 matplotlib 和 numpy，并实现更丰富的视觉效果：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation

class Particle:
    def __init__(self, x, y, vx, vy, color):
        self.x = x
        self.y = y
        self.vx = vx
        self.vy = vy
        self.color = color
        self.life = np.random.randint(50, 100)  # 生命值，控制粒子的存活时间

    def update(self):
        self.x += self.vx
        self.y += self.vy
        self.vy -= 0.05  # 重力影响
        self.life -= 1  # 每次更新生命值减少

    def is_alive(self):
        return self.life > 0

class Firework:
    def __init__(self, x, y):
        self.x = x
        self.y = y
        self.particles = []
        self.exploded = False

    def explode(self):
        if not self.exploded:
            num_particles = np.random.randint(100, 200)
            angles = np.linspace(0, 2 * np.pi, num_particles)
            speeds = np.random.uniform(1, 4, num_particles)
            colors = plt.cm.hsv(np.random.rand(num_particles))  # 使用HSV颜色

            for angle, speed, color in zip(angles, speeds, colors):
                vx = speed * np.cos(angle)
                vy = speed * np.sin(angle)
                self.particles.append(Particle(self.x, self.y, vx, vy, color))
            self.exploded = True

    def update(self):
        if self.exploded:
            for particle in self.particles:
                particle.update()

    def get_particles(self):
        return [p for p in self.particles if p.is_alive()]

# 初始化画布
fig, ax = plt.subplots()
ax.set_xlim(0, 10)
ax.set_ylim(0, 10)
ax.set_facecolor('black')

fireworks = []

# 生成烟花
def generate_fireworks(num):
    for _ in range(num):
        x = np.random.uniform(1, 9)
        y = 0  # 从底部开始
        firework = Firework(x, y)
        fireworks.append(firework)

generate_fireworks(3)

# 动画更新函数
def update(frame):
    ax.clear()
    ax.set_xlim(0, 10)
    ax.set_ylim(0, 10)
    ax.set_facecolor('black')

    # 处理烟花升起
    for firework in fireworks:
        if firework.y < 8:  # 设置升起的高度
            firework.y += 0.1
        else:
            firework.explode()  # 当升至最高点，爆炸
            firework.update()  # 更新粒子

        # 绘制存活的粒子
        particles = firework.get_particles()
        for particle in particles:
            ax.scatter(particle.x, particle.y, color=particle.color, s=10)

# 创建动画
ani = animation.FuncAnimation(fig, update, frames=100, interval=50)
plt.show()