用python在环面上绘制动画？

我最近编写了一个代码，可以模拟粒子在周期性空间中的运动。具体来说，它产生了一个二维图，图中有N个点（N~10^4），最初聚集在中心，然后根据匹配的速度展开。由于它是一个周期性空间，任何超出上限的点实际上都会回到下限，反之亦然。为了进行说明，以下是两幅图像：

初始位置

一段时间后

每个点应水平移动，向右或向左（分别为正速度或负速度）

我用Python编程，但现在，在我的项目范围内，我想在圆环上模拟同样的东西。为了让您更好地了解它的外观，请看下面的图片：

从矩形到圆环体的变换

（假设我的初始二维平面是初始矩形，我想将其转换为最终圆环体）

因此，在这种情况下，我们会看到每个粒子在环面上移动。前一张图片将对应于在圆环的“单个”圆环上聚集的粒子，前一张图片将对应于“填充”圆环的整个表面

由于我以前的模拟代码是用Python编写的，我想知道我是否仍然可以将其用于此任务。如果是的话，我想知道一些关于如何做的线索，否则，什么是最好的语言

---

我希望我已经说清楚了。对于我用英语可能会犯的一些错误，我提前表示歉意。

您将如何在2D空间中表示您的圆环体？你会想象一个3D绘图，还是会计算每个粒子沿其“圆环空间”的路径，然后以某种方式将该运动投影到一个正方形的2D窗口？@VegardKT谢谢你的回答。我认为3D绘图更好，这里的图片只是为了显示变换，这样你就可以可视化粒子在环面上的行为。你需要的是使用t将环面上点的2D位置映射到3D空间，然后使用pythonSo创建3D绘图。我想你的问题是；“如何计算圆环体上的粒子路径”@bracco23所以我会首先在2D绘图上绘制所有位置，然后使用其方程在3D圆环体上绘制匹配位置？