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')