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")
|
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"))
|
||||||
|
|
Loading…
Reference in a new issue