traj[:,:,:,:,k] = tmp
k+=1
# SCATTER PLOT
nrows = 2
ncols = 2
left_space = 0.08
right_space = 0.02
bottom_space = 0.05
top_space = 0.05
hor_space = 0.
ver_space = 0.05
fig_width = 8.27
ax = square_grid_plot(nrows,ncols,left_space,right_space,bottom_space,top_space,hor_space,ver_space,fig_width)
k=0
read_index = [0,2,1,3]
for k in range(4):
ax[k].scatter(traj[bin_no,0,:,0,read_index[k]],traj[bin_no,1,:,0,read_index[k]],c = 'k',s = 2)
ax[k].set_title('Time = %i days' % int(time_read[read_index[k]]))
ax[k].set_ylim([0,512])
ax[k].set_xlim([0,512])
ax[0].set_ylabel('Y (km)')
ax[1].set_ylabel('Y (km)')
ax[1].set_xlabel('X (km)')
ax[3].set_xlabel('X (km)')
for p in range(len(bin_index[i, j, :]))
if bin_index[i, j, p] == b + 1
]
spread_new[i, j, b] = np.mean(x_dispersion_new)
traj = np.reshape(traj, [2, 2, nbins * npoints, ndims, nt])
left_space = 0.08
right_space = 0.05
bottom_space = 0.08
top_space = 0.05
hor_space = 0.05
ver_space = 0.05
fig_width = 8.27
ax = square_grid_plot(2, 2, left_space, right_space, bottom_space, top_space,
hor_space, ver_space, fig_width)
k = 0
for i in range(2):
for j in range(2):
ax[k].scatter(np.transpose(traj[i, j, :, 0, -1]),
np.transpose(traj[i, j, :, 1, -1]))
ax[k].set_xlim([-4000, 6000])
ax[k].grid()
k += 1
ax[0].set_title('Coherent Jet')
ax[2].set_title('Latent Jet')
ax[0].set_ylabel('Top Layer')
ax[1].set_ylabel('Bottom Layer')
ax[1].set_xlabel('Stokes Drift')
ax[3].set_xlabel('Stokes Drift')
