python爱心粒子特效代码三维动态要求两种颜色及以上实现

要实现Python的三维爱心粒子特效并支持多种颜色，我们可以使用Pygame库来创建一个简单的窗口环境，结合随机生成的颜色和运动轨迹来模拟爱心粒子效果。以下是一个简化的示例代码：

import pygame
import random

# 初始化pygame
pygame.init()

# 设置窗口大小
window_size = (800, 600)
screen = pygame.display.set_mode(window_size)

# 定义颜色列表
colors = [(255, 0, 0), (0, 255, 0), (0, 0, 255)]  # 你可以添加更多颜色

class HeartParticle:
    def __init__(self, pos, speed=(1, 1)):
        self.pos = pos
        self.speed = speed
        self.color = random.choice(colors)

    def update(self):
        self.pos = (self.pos[0] + self.speed[0], self.pos[1] + self.speed[1])
        if self.pos[0] < 0 or self.pos[0] >= window_size[0]:
            self.speed = (-self.speed[0], self.speed[1])  # 当粒子碰到边界时反弹
        if self.pos[1] < 0 or self.pos[1] >= window_size[1]:
            self.speed = (self.speed[0], -self.speed[1])

    def render(self):
        pygame.draw.circle(screen, self.color, self.pos, radius=2)  # 画圆点表示粒子

# 创建大量爱心粒子
particle_count = 50
particles = [HeartParticle((random.randint(0, window_size[0]), random.randint(0, window_size[1])) for _ in range(particle_count)]

# 游戏主循环
running = True
while running:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False

    screen.fill((0, 0, 0))  # 清空屏幕
    for particle in particles:
        particle.update()
        particle.render()

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

# 清理pygame资源
pygame.quit()

在这个例子中，我们创建了一个`HeartParticle`类，包含位置、速度和颜色属性。粒子会在屏幕上随机出现，每次更新时移动并改变颜色（如果颜色用完则重新选择）。当粒子到达边缘时，它们会被反弹回对称的位置。