If you replace this line of code in your second script:

timesteps, nx, ny = hraw.shape

with this example data (you have to use your own!)

timesteps, nx, ny = 174, 200, 50 
hraw = np.random.rand(timesteps, nx, ny)  # Example-horizontal data 
uraw = np.random.rand(timesteps, nx, ny)  # Example-speed data

Looks it what you search for?