python绘制春节烟花的示例代码

import pygame
import random
import math
from pygame.locals import *
 
# 初始化pygame
pygame.init()
 
# 设置屏幕大小
screen_width = 800
screen_height = 600
screen = pygame.display.set_mode((screen_width, screen_height))
 
# 设置标题
pygame.display.set_caption('春节烟花')
 
# 定义烟花参数
firework_speed = 5
firework_radius = 2
firework_explosion_radius = 60
colors = [
    (255, 0, 0),  # Red
    (0, 255, 0),  # Green
    (0, 0, 255),  # Blue
    (255, 255, 0),  # Yellow
    (255, 165, 0),  # Orange
    (255, 255, 255)  # White
]
 
# 定义Firework类
class Firework:
    def __init__(self, x, y, color, exploded=False):
        self.x = x
        self.y = y
        self.color = color
        self.exploded = exploded
        self.particles = []
 
    def move(self):
        if not self.exploded:
            self.y -= firework_speed
 
    def explode(self):
        if not self.exploded:
            for angle in range(0, 360, 5):
                dir_x = math.cos(math.radians(angle))
                dir_y = math.sin(math.radians(angle))
                self.particles.append((self.x, self.y, dir_x, dir_y, self.color))
            self.exploded = True
 
    def update(self):
        if self.exploded:
            for particle in self.particles:
                index = self.particles.index(particle)
                particle_x, particle_y, dir_x, dir_y, color = particle
                particle_x += dir_x * 2
                particle_y += dir_y * 2
                self.particles[index] = (particle_x, particle_y, dir_x, dir_y, color)
                if self.distance(particle_x, particle_y) > firework_explosion_radius:
                    self.particles.pop(index)
 
    def show(self):
        if not self.exploded:
            pygame.draw.circle(screen, self.color, (int(self.x), int(self.y)), firework_radius)
        else:
            for particle in self.particles:
                particle_x, particle_y, dir_x, dir_y, color = particle
                pygame.draw.circle(screen, color, (int(particle_x), int(particle_y)), firework_radius)
    
    def distance(self, x, y):
        return math.sqrt((self.x - x) ** 2 + (self.y - y) ** 2)
 
fireworks = [Firework(random.randint(0, screen_width), screen_height - 10, random.choice(colors))]
 
# 游戏主循环
running = True
while running:
    screen.fill((0, 0, 0))  # use a dark sky background
 
    # 执行事件循环
    for event in pygame.event.get():
        if event.type == QUIT:
            running = False
 
    # 更新和显示烟花
    for firework in fireworks:
        if not firework.exploded and firework.y < screen_height / 2 + random.randint(-100, 100):
            firework.explode()
        firework.move()
        firework.update()
        firework.show()
 
    # 随机发射新的烟花
    if random.randint(0, 20) == 1:
        fireworks.append(Firework(random.randint(0, screen_width), screen_height - 10, random.choice(colors)))
 
    # 删除已完成的烟花
    for firework in fireworks:
        if firework.exploded and len(firework.particles) == 0:
            fireworks.remove(firework)
 
    pygame.display.flip()
    pygame.time.Clock().tick(30)  # 控制游戏最大帧率为30fps
 
pygame.quit()

python -m pip install --upgrade pip

python -m pip install pygame

python -c "import pygame; print(pygame.ver)"

import turtle
import random
 
# 设置屏幕背景
screen = turtle.Screen()
screen.bgcolor("black")
 
# 创建烟花的绘图对象
firework = turtle.Turtle()
firework.speed(0)
firework.hideturtle()
 
# 绘制烟花的方法
def draw_firework():
    colors = ["red", "yellow", "blue", "green", "orange", "purple", "white"]
    # 烟花升空
    firework.penup()
    firework.goto(random.randint(-200, 200), random.randint(-200, 0))
    firework.pendown()
    
    # 烟花爆炸
    explode_times = random.randint(5, 15)
    for i in range(explode_times):
        firework.color(random.choice(colors))
        firework.pensize(random.randint(1, 5))
        firework.speed(0)
        angle = 360 / explode_times
        firework.seth(i * angle)
        firework.forward(random.randint(20, 150))
        firework.backward(random.randint(20, 150))
 
# 重复绘制烟花
for _ in range(10):
    draw_firework()
 
# 点击屏幕后退出
screen.exitonclick()