小编ma7*_*642的帖子

我正在尝试使用 Matplotlib 制作一个简单的动画来检查我的模拟是否运行良好。我想看看两个粒子如何在 xy 平面上移动：如果代码有效，它们应该吸引并结束在同一点或在空间中接近。

我计算“for”循环内粒子的位置，每次获得位置时，我都会使用 plt.scatter(x, y) 绘制它们，其中 x 和 y 是时间 t = t + dt 时的位置。我在网上读到我可以在循环内编写 plt.pause(0.05) ，这将产生我正在寻找的简单动画。

我的代码示例如下：

import numpy as np
import matplotlib.pyplot as plt

for k in range(steps):
    pos = computeNewPos(pos, vel, force)        

    plt.scatter(pos[0, 0], pos[0, 1], label = '1', color = 'r')
    plt.scatter(pos[1, 0], pos[1, 1], label = '2', color = 'b')

    plt.xlabel('X')
    plt.ylabel('Y')

    plt.pause(0.05)

plt.show()

这可行，但我得到的“动画”包含旧数据点，我只想看到更新的位置。这将使跟踪粒子的位置变得更加容易。如何“擦除”for 循环内每次运行的旧点？有没有办法在每次运行时清除框架？