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
1 changed files with 2 additions and 2 deletions

View File

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