新年到，赶紧用Python安排一场烟花秀

本篇文章，我们带给大家一个小案例：用 Python 制作一个炫酷烟花秀，祝大家新年快乐，顺便也提前送上春节祝福。

开始之前先看一下最终效果：

环境介绍

• 语言：Python
• 库：Pygame

原理介绍

在介绍代码之前，先介绍下 Pygame 绘制烟花的基本原理，烟花从发射到绽放一共分为三个阶段：

1，发射阶段：在这一阶段烟花的形状是线性向上，通过设定一组大小不同、颜色不同的点来模拟“向上发射” 的运动运动，运动过程中 5个点被赋予不同大小的加速度，随着时间推移，后面的点会赶上前面的点，最终所有点会汇聚在一起，处于 绽放准备阶段；

2，烟花绽放：烟花绽放这个阶段，是由一个点分散多个点向不同方向发散，并且每个点的移动轨迹可需要被记录，目的是为了追踪整个绽放轨迹。

3，烟花凋零，此阶段负责描绘绽放后烟花的效果，绽放后的烟花，而在每一时刻点的下降速度和亮度（代码中也叫透明度）是不一样的，因此在代码里，将烟花绽放后将每个点赋予两个属性：分别为重力向量和生命周期，来模拟烟花在不同时期时不同的展现效果。

代码实操

代码部分将烟花封装为三个类：

Firework : 烟花整体；

Particle：烟花粒子（包含轨迹）

Trail：烟花轨迹，本质上是一个点 。

三个类之间的关系为：一个Firework 由多个 Particle 构成，而一个 Particle 由多个 Trail 构成

首先设置全局变量，例如重力向量，窗口大小，Trail 的颜色列表（多为灰色或白色）以及不同状态下 Trail 之间间隔

gravity = vector(0, 0.3)
DISPLAY_WIDTH = DISPLAY_HEIGHT = 800


trail_colours = [(45, 45, 45), (60, 60, 60), (75, 75, 75), (125, 125, 125), (150, 150, 150)]
dynamic_offset = 1
static_offset = 3

创建 Trail 类，定义 show 方法绘制轨迹 、get_pos 实时获取轨迹坐标

class Trail:

    def __init__(self, n, size, dyna