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

Python 烟花展示：使用 Pygame 创建绚丽的夜空

news 来源：原创 2024/9/28 1:19:22

在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-第一篇+源码讲解

在这里插入图片描述