1
Fork 0

Add support for arbitrary number of layers

This commit is contained in:
Edgar P. Burkhart 2022-03-30 12:16:19 +02:00
parent 1c59e24cea
commit 858928353a
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227

View file

@ -55,7 +55,7 @@ vk = np.sqrt((velk[n] ** 2).sum(axis=1))
lines = ax.plot(x, zk[n].T, c="#0066cc") lines = ax.plot(x, zk[n].T, c="#0066cc")
quiv = [] quiv = []
for i in range(10): for i in range(len(lines)-1):
quiv.append( quiv.append(
ax.quiver( ax.quiver(
x[::50], x[::50],
@ -87,4 +87,4 @@ ani = animation.FuncAnimation(
fig, animate, frames=wl[:, 0].size, interval=20, blit=True fig, animate, frames=wl[:, 0].size, interval=20, blit=True
) )
ani.save(out.joinpath("layers.mp4", codec="h265")) ani.save(out.joinpath("layers.mp4"))