Add support for arbitrary number of layers
This commit is contained in:
parent
1c59e24cea
commit
858928353a
1 changed files with 2 additions and 2 deletions
|
@ -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"))
|
||||
|
|
Loading…
Reference in a new issue