1
Fork 0

Removed reshape in layers

This commit is contained in:
Edgar P. Burkhart 2022-03-17 11:57:29 +01:00
parent fbf31bce48
commit 8863cdff8d
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227

View file

@ -50,7 +50,6 @@ fig, ax = plt.subplots()
# )
n = 0
velk = velk.reshape((6001, 10, 2, 1251))
vk = np.sqrt((velk[n] ** 2).sum(axis=1))
# print(vk.shape)
# plt.imshow(vk)
@ -83,6 +82,7 @@ def animate(k):
)
for i, l in enumerate(lines):
l.set_ydata(zk[k, i])
return *quiv, *lines
ani = animation.FuncAnimation(