python圣诞树代码动态3d粒子

在Python中，可以使用Pygame库以及一些基础的数学知识来创建一个动态的3D粒子效果，模拟圣诞树。下面是一个简单的示例，展示了如何通过绘制几何形状并改变其位置和颜色来制作一个基本的3D“粒子”圣诞树：

import pygame
from pygame.locals import *
import math

# 初始化pygame
pygame.init()

# 设置窗口大小
window_width = 800
window_height = 600
screen = pygame.display.set_mode((window_width, window_height))
pygame.display.set_caption("3D Particle Christmas Tree")

# 定义颜色和速度范围
color_range = [(255, 0, 0), (0, 255, 0), (0, 0, 255)]
particle_speed = [math.pi / 4 for _ in range(3)]

def create_particle():
    pos = (random.randint(0, window_width), random.randint(0, window_height))
    velocity = (
        particle_speed[0] * random.choice([-1, 1]),
        particle_speed[1] * random.choice([-1, 1]),
        particle_speed[2] * random.choice([-1, 1])
    )
    return pos, velocity

# 主循环
running = True
while running:
    # 处理事件
    for event in pygame.event.get():
        if event.type == QUIT or (event.type == KEYDOWN and event.key == K_ESCAPE):
            running = False

    # 清除屏幕
    screen.fill((0, 0, 0))

    # 创建新粒子或更新现有粒子的位置
    particles = []
    while len(particles) < 10:  # 添加最多10个粒子
        pos, velocity = create_particle()
        particles.append((pos, velocity))

    for particle in particles:
        pos, vel = particle
        pos = tuple(int(i + vel[i]) for i in range(len(pos)))
        if pos[0] <= 0 or pos[0] >= window_width or pos[1] <= 0 or pos[1] >= window_height:
            particles.remove(particle)
        else:
            color = color_range[random.randint(0, len(color_range) - 1)]
            pygame.draw.polygon(screen, color, ((pos[0], pos[1]), (pos[0] + 5, pos[1] + 5), (pos[0] + 5, pos[1] - 5)))

    # 更新屏幕
    pygame.display.update()

# 结束游戏
pygame.quit()

这个代码会生成一些随机移动的红、绿、蓝小点，看起来像粒子一样组成一个简单3D圣诞树的效果。注意这只是一个基础版本，你可以进一步增加复杂度，比如使用纹理、旋转等。