用python写一个3D悬浮粒子爱心程序

在Python中，我们可以利用matplotlib库创建一个3D空间中的悬浮爱心形状粒子动画。下面是一个简单的例子，使用了mplot3d模块：

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
import random

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 创建爱心的顶点坐标
vertices = [
    [-0.5, -0.5, 0],
    [0.5, -0.5, 0],
    [0, 0.5, 0],
    [-0.5, -0.5, 0.5],
    [0.5, -0.5, 0.5],
    [0, 0.5, 0.5]
]

# 定义颜色和大小
colors = ['r', 'g', 'b', 'y', 'c', 'm']
sizes = [random.uniform(0.1, 0.3) for _ in range(len(vertices))]

# 动画函数
def animate(i):
    # 更新每个粒子的位置
    for j, vertex in enumerate(vertices):
        x, y, z = vertex
        ax.scatter(x, y, z, c=colors[j], s=sizes[j])

# 创建动画
ani = animation.FuncAnimation(fig, animate, frames=np.arange(100), interval=50)

plt.show()