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

devtools/2024/9/22 13:00:56/

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

示例代码

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

python">import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animationclass Particle:def __init__(self, x, y, vx, vy, color):self.x = xself.y = yself.vx = vxself.vy = vyself.color = colorself.life = np.random.randint(50, 100)  # 生命值，控制粒子的存活时间def update(self):self.x += self.vxself.y += self.vyself.vy -= 0.05  # 重力影响self.life -= 1  # 每次更新生命值减少def is_alive(self):return self.life > 0class Firework:def __init__(self, x, y):self.x = xself.y = yself.particles = []self.exploded = Falsedef 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 = Truedef 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.1else: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()