在Python中，使用pygame库可以轻松地创建图形和动画效果，非常适合制作各种游戏和视觉展示。今天，我们将一起探索如何使用pygame来制作一个简单的烟花展示程序。这个程序将模拟烟花在夜空中绽放的壮丽景象，通过随机生成的粒子来模拟烟花的爆炸效果。

准备工作

首先，确保你已经安装了pygame库。如果还没有安装，可以通过pip命令来安装：

pip install pygame

导入必要的库

import pygame  # 导入pygame库，用于处理图形和事件  
import random  # 导入random库，用于生成随机数  
import math  # 导入math库，用于数学运算  
import sys  # 导入sys库，用于控制程序的退出

初始化pygame和设置窗口

# 初始化pygame  
pygame.init()  
  
# 设置窗口  
WIDTH, HEIGHT = 640, 480  # 设置窗口的宽度和高度  
screen = pygame.display.set_mode((WIDTH, HEIGHT))  # 创建一个窗口对象  
pygame.display.set_caption('Python 烟花展示')  # 设置窗口的标题

定义颜色和粒子类

# 定义新的颜色列表，并打乱它  
COLORS = [(255, 165, 0), (128, 0, 128), (0, 255, 127), (255, 140, 0), (0, 191, 255), (255, 32, 128), (64, 224, 208)]  
random.shuffle(COLORS)  
  
# 定义粒子类  
class Particle:  
    def __init__(self, x, y, dx, dy, color, size=3, alpha=255):  
        self.x = x  # 粒子的x坐标  
        self.y = y  # 粒子的y坐标  
        self.dx = dx  # 粒子在x方向上的速度  
        self.dy = dy  # 粒子在y方向上的速度  
        self.color = color  # 粒子的颜色（RGB格式）  
        self.size = size  # 粒子的大小  
        self.alpha = alpha  # 粒子的透明度  
  
    def update(self):  
        self.x += self.dx  # 更新粒子的x坐标  
        self.y += self.dy  # 更新粒子的y坐标  
        self.dy += 0.1  # 粒子受到重力的影响，y方向速度增加  
        self.alpha -= 5  # 粒子的透明度逐渐降低  
  
    def draw(self, surface):  
        # 注意：pygame的draw函数不支持直接的RGBA颜色，所以这里仅使用RGB并忽略alpha值，通过透明度处理实现淡出效果  
        pygame.draw.circle(surface, (self.color[0], self.color[1], self.color[2]), (int(self.x), int(self.y)), self.size)

定义烟花类

# 定义烟花类  
class Firework:  
    def __init__(self, x, y, color, particle_count=50):  
        self.x = x  # 烟花的x坐标  
        self.y = y  # 烟花的y坐标  
        self.color = color  # 烟花的颜色  
        self.particles = [self.create_particle() for _ in range(particle_count)]  # 初始化烟花的粒子列表  
  
    def create_particle(self):  
        angle = random.uniform(0, 2 * math.pi)  # 随机生成一个角度  
        speed = random.uniform(1, 6)  # 随机生成一个速度  
        dx = speed * math.cos(angle)  # 计算x方向上的速度分量  
        dy = speed * math.sin(angle)  # 计算y方向上的速度分量  
        size = random.randint(2, 6)  # 随机生成粒子的大小  
        return Particle(self.x, self.y, dx, dy, self.color, size)  
  
    def update(self):  
        self.particles = [p for p in self.particles if p.alpha > 0]  # 过滤出仍然可见的粒子  
        for particle in self.particles:  
            particle.update()  # 更新每个粒子的状态  
  
    def draw(self):  
        for particle in self.particles:  
            particle.draw(screen)  # 将粒子绘制到屏幕上

主循环和烟花生成

# 烟花列表  
fireworks = []  
  
# 主循环  
clock = pygame.time.Clock()  
running = True  
while running:  
    for event in pygame.event.get():  
        if event.type == pygame.QUIT:  
            running = False  
  
    screen.fill((0, 0, 0))  # 清屏  
    if random.random() < 0.05:  # 以较低的概率生成新的烟花  
        x = random.randint(0, WIDTH)  
        y = random.randint(HEIGHT // 2, HEIGHT)  
        color = random.choice(COLORS)  
        fireworks.append(Firework(x, y, color, particle_count=random.randint(30, 70)))  
  
    for firework in fireworks[:]:  # 遍历烟花列表的副本  
        firework.update()  
        if not firework.particles:  
            fireworks.remove(firework)  
        firework.draw()  
  
    pygame.display.flip()  # 更新屏幕  
    clock.tick(60)  # 设置帧率为60帧/秒  
  
pygame.quit()  # 退出pygame  
sys.exit()  # 退出程序

完整代码

import pygame  # 导入pygame库，用于处理图形和事件  
import random  # 导入random库，用于生成随机数  
import math  # 导入math库，用于数学运算  
import sys  # 导入sys库，用于控制程序的退出  
  
# 初始化pygame  
pygame.init()  # 初始化pygame模块，为使用pygame功能做准备  
  
# 设置窗口  
WIDTH, HEIGHT = 640, 480  # 设置窗口的宽度和高度  
screen = pygame.display.set_mode((WIDTH, HEIGHT))  # 创建一个窗口对象  
pygame.display.set_caption('Python 烟花展示')  # 设置窗口的标题  
  
# 定义新的颜色列表，并打乱它  
COLORS = [(255, 165, 0), (128, 0, 128), (0, 255, 127), (255, 140, 0), (0, 191, 255), (255, 32, 128), (64, 224, 208)]  
random.shuffle(COLORS)  # 使用random.shuffle函数打乱颜色列表  
  
# 定义粒子类  
class Particle:  
    def __init__(self, x, y, dx, dy, color, size=3, alpha=255):  
        """初始化粒子对象，包括位置、速度、颜色、大小和透明度"""  
        self.x = x  # 粒子的x坐标  
        self.y = y  # 粒子的y坐标  
        self.dx = dx  # 粒子在x方向上的速度  
        self.dy = dy  # 粒子在y方向上的速度  
        self.color = color  # 粒子的颜色（RGBA格式，但这里只使用RGB）  
        self.size = size  # 粒子的大小  
        self.alpha = alpha  # 粒子的透明度  
  
    def update(self):  
        """更新粒子的位置和透明度"""  
        self.x += self.dx  # 更新粒子的x坐标  
        self.y += self.dy  # 更新粒子的y坐标  
        self.dy += 0.1  # 粒子受到重力的影响，y方向速度增加  
        self.alpha -= 5  # 粒子的透明度逐渐降低  
  
    def draw(self, surface):  
        """绘制粒子到指定的表面上，考虑透明度"""  
        # 注意：pygame的draw函数不支持直接的RGBA颜色，所以这里仅使用RGB并忽略alpha值，通过透明度处理实现淡出效果  
        pygame.draw.circle(surface, (self.color[0], self.color[1], self.color[2]), (int(self.x), int(self.y)), self.size)  
  
# 定义烟花类  
class Firework:  
    def __init__(self, x, y, color, particle_count=50):  
        """初始化烟花对象，包括位置、颜色、粒子数量，并创建初始粒子"""  
        self.x = x  # 烟花的x坐标  
        self.y = y  # 烟花的y坐标  
        self.color = color  # 烟花的颜色  
        self.particles = [self.create_particle() for _ in range(particle_count)]  # 初始化烟花的粒子列表  
  
    def create_particle(self):  
        """创建一个新的粒子，随机确定其初始速度和方向"""  
        angle = random.uniform(0, 2 * math.pi)  # 随机生成一个角度  
        speed = random.uniform(1, 6)  # 随机生成一个速度  
        dx = speed * math.cos(angle)  # 计算x方向上的速度分量  
        dy = speed * math.sin(angle)  # 计算y方向上的速度分量  
        size = random.randint(2, 6)  # 随机生成粒子的大小  
        return Particle(self.x, self.y, dx, dy, self.color, size)  
  
    def update(self):  
        """更新烟花的所有粒子，并移除已经完全透明的粒子"""  
        self.particles = [p for p in self.particles if p.alpha > 0]  # 过滤出仍然可见的粒子  
        for particle in self.particles:  
            particle.update()  # 更新每个粒子的状态  
  
    def draw(self):  
        """绘制烟花的所有粒子"""  
        for particle in self.particles:  
            particle.draw(screen)  # 将粒子绘制到屏幕上  
  
# 烟花列表  
fireworks = []  
  
# 主循环  
clock = pygame.time.Clock()  # 创建一个Clock对象用于控制帧率  
running = True  # 控制主循环的标志位  
while running:  
    for event in pygame.event.get():  # 遍历所有事件  
        if event.type == pygame.QUIT:  # 如果事件是关闭窗口  
            running = False  # 退出主循环  
  
    screen.fill((0, 0, 0))  # 用黑色填充整个屏幕，清屏  
    if random.random() < 0.05:  # 以较低的概率生成新的烟花  
        x = random.randint(0, WIDTH)  # 随机生成烟花的x坐标  
        y = random.randint(HEIGHT // 2, HEIGHT)  # 随机生成烟花的y坐标（从屏幕中间偏上到顶部）  
        color = random.choice(COLORS)  # 从打乱后的颜色列表中随机选择一个颜色  
        fireworks.append(Firework(x, y, color, particle_count=random.randint(30, 70)))  # 创建一个新的烟花对象并添加到列表中  
  
    for firework in fireworks[:]:  # 遍历烟花列表的副本（避免在迭代过程中修改列表）  
        firework.update()  # 更新烟花的状态  
        if not firework.particles:  # 如果烟花的粒子列表为空（即所有粒子都已消失）  
            fireworks.remove(firework)  # 从列表中移除该烟花  
        firework.draw()  # 绘制烟花  
  
    pygame.display.flip()  # 更新整个屏幕的内容  
    clock.tick(60)  # 设置帧率为60帧/秒  
  
pygame.quit()  # 退出pygame  
sys.exit()  # 退出程序

推荐阅读

Python基础

Python全网最全基础课程笔记(一)——基础入门
Python全网最全基础课程笔记(二)——变量
Python全网最全基础课程笔记(三)——所有运算符+运算符优先级
Python全网最全基础课程笔记(四)——基本数据类型
Python全网最全基础课程笔记(五)——选择结构+Python新特性Match
Python全网最全基础课程笔记(六)——循环结构
Python全网最全基础课程笔记(七)——列表，跟着思维导图和图文来学习，爆肝2w字，无数代码案例！
Python全网最全基础课程笔记(八)——字典，跟着思维导图和图文来学习，爆肝2w字，无数代码案例！
Python全网最全基础课程笔记(九)——集合，跟着思维导图和图文来学习，爆肝2w字，无数代码案例！
Python全网最全基础课程笔记(十)——元组，跟着思维导图和图文来学习，爆肝2w字，无数代码案例！
Python全网最全基础课程笔记(十一)——字符串所有操作，跟着思维导图和图文来学习，爆肝2w字，无数代码案例！
Python全网最全基础课程笔记(十二)——函数，跟着思维导图和图文来学习，爆肝2w字，无数代码案例！
Python全网最全基础课程笔记(十三)——作用域，跟着思维导图和图文来学习，爆肝2w字，无数代码案例！

Flink入门到就业

2024年最新Flink教程,从基础到就业，大家一起学习–基础篇
2024年最新Flink教程,从基础到就业，大家一起学习–入门篇
2024年最新Flink教程,从基础到就业，大家一起学习–Flink集群部署
2024年最新Flink教程,从基础到就业，大家一起学习–flink部署和集群部署(从本地测试到公司生产环境如何部署项目源码)
2024年最新Flink教程,从基础到就业，大家一起学习–Flink运行架构底层源码详解+实战
2024年最新Flink教程,从基础到就业，大家一起学习–Flink DataStream API-第一篇+源码讲解