写一个python动态粒子圣诞树代码

在Python中，我们可以利用pygame库创建一个简单的动态粒子圣诞树效果。这里是一个简化的例子，展示如何使用基本的形状和颜色变化来模拟圣诞树的动态效果：

import pygame
import random

# 初始化pygame
pygame.init()

# 设置窗口大小和背景色
win_size = (800, 600)
screen = pygame.display.set_mode(win_size)
bg_color = (0, 0, 0)

# 定义圣诞树的基本形状
def draw_tree(branch_width, color):
    for y in range(-branch_width, branch_width + 1):
            dx = abs(x) + abs(y)
            dy = min(abs(x), abs(y))
            distance = sqrt(dx ** 2 + dy ** 2)
            if distance < branch_width / 4:
                screen.set_at((x + win_size[0] // 2, y + win_size[1] // 2), color)

# 粒子属性
particle_num = 50
particle_size = 5
colors = [(255, 255, 255)] * particle_num  # 白色粒子
speeds = [random.uniform(0.1, 0.3) for _ in range(particle_num)]

while True:
    # 清除屏幕
    screen.fill(bg_color)

    # 更新粒子位置
    for i in range(particle_num):
        particles[i].y += speeds[i]
        if particles[i].y > win_size[1]:
            particles[i].y = -particle_size
            colors[i] = (255, 255, 255)  # 当粒子下落到底部时变回白色

    # 绘制圣诞树和粒子
    draw_tree(particle_size, colors[random.randint(0, particle_num - 1)])
    for particle in particles:
        pygame.draw.circle(screen, colors[particle.index], (particle.x, particle.y), particle_size)

    # 检查事件并更新
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            quit()

    # 更新显示
    pygame.display.update()

在这个代码中，我们使用了pygame的`set_at`函数绘制像素点形成圣诞树的分支，并通过改变粒子的位置和颜色来模拟动态效果。注意这只是一个基础示例，如果你想添加更复杂的动画效果，比如旋转、闪烁等，可以进一步扩展。