Python 中的散点动画

Question

我正在尝试在Python 3.6（在Spyder IDE中）中对散点图进行动画处理，有几个教程/示例（a、b、c等），但是我的更新功能不知何故收到一组与我认为我提供的坐标非常不同的坐标。

以下是简化的示例代码，其中 4 个点向中心延伸。我将其绘制两次，第一次作为线条，第二次作为动画。线图看起来正确，但动画在点的位置及其轨迹方面显示了非常不同的行为。我推断我错误地向更新函数提供了坐标信息，但我似乎无法确定原因/方式。

我可以提供数据来显示发生了什么与应该发生什么，尽管为了简洁起见，我现在只是提供这个示例代码。

import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt
import matplotlib.animation as anim

npt=4
ntime=101
solx=np.zeros((npt,ntime))
soly=np.zeros((npt,ntime))
solx[0,0:]=np.linspace(0,40,ntime)
solx[1,0:]=solx[0,0:];
solx[2,0:]=100-solx[0,0:];
solx[3,0:]=solx[2,0:]
soly[0,0:]=np.linspace(0,80,ntime)
soly[1,0:]=200-soly[0,0:]
soly[2,0:]=soly[0,0:]
soly[3,0:]=soly[1,0:]


#%% Plots
colord=mpl.cm.gist_ncar(np.linspace(0,0.9,npt))
of2,oa2=plt.subplots(1)
for cnt in range(0,npt): oa2.plot(solx[cnt,0:],soly[cnt,0:],c=colord[cnt])

of3,oa3=plt.subplots()
osc3=oa3.scatter(solx[0:,0],soly[0:,0],c=colord,marker='+')
def aninit():
    osc3.set_offsets(np.stack((solx[0:,0],soly[0:,0]),axis=0))
    return osc3,
def aniupdate(i):
    osc3.set_offsets(np.stack((solx[0:,i],soly[0:,i]),axis=0))
    return osc3,

ani=anim.FuncAnimation(of3,aniupdate,init_func=aninit,frames=range(0,ntime),interval=100,blit=True,repeat=True,repeat_delay=1000)
plt.show()

Answer 1

如果更换

np.stack((solx[0:,0],soly[0:,0]),axis=0)
np.stack((solx[0:,i],soly[0:,i]),axis=0)

与：

list(zip(solx[0:,0],soly[0:,0]))
list(zip(solx[0:,i],soly[0:,i]))

或者（如果你想使用 numpy 数组而不是列表：

np.array([solx[0:,0],soly[0:,0]]).T
np.array([solx[0:,i],soly[0:,i]]).T

在

aninit()

和

aniupdate(i)

函数中，行为应符合预期。

这是我用来制作绘图的代码：

def aninit():
    offsetlist = np.array([solx[0:,0],soly[0:,0]]).T
    osc3.set_offsets(offsetlist)
    return osc3,

def aniupdate(i):
    offsetlist = np.array([solx[0:,i],soly[0:,i]]).T
    osc3.set_offsets(offsetlist)
    return osc3,

参考资料：

• 下雨模拟
• old_animation示例代码：dynamic_collection.py