diff --git a/plots/animation.py b/plots/animation.py index d52e2d7464d9674caa3c703b790287644091ab3f..ad7ab8561e43cfd9094a2ae59db6bf56a26b99e3 100644 --- a/plots/animation.py +++ b/plots/animation.py @@ -20,6 +20,7 @@ def animate(i): yi = y[i*N:(i+1)*N] plt.cla() plt.scatter(xi, yi, s=0.1) + plt.scatter(np.sum(xi)/N, np.sum(yi)/N, s=1, c="red") plt.xlabel("x") plt.ylabel("y") plt.axis("equal")