Commit 86cb7ee8 by Veit Lüschow

### about to iniotialize density in geostrophic balance

parent d1b0be36
 ... @@ -64,7 +64,7 @@ def solve_streamfunction(vs): ... @@ -64,7 +64,7 @@ def solve_streamfunction(vs): line_forc = allocate(vs, ('isle',)) line_forc = allocate(vs, ('isle',)) if vs.nisle > 1: if vs.nisle > 1: # calculate island integrals of forcing, keep psi constant on island 1 # ulate island integrals of forcing, keep psi constant on island 1 line_forc[1:] = utilities.line_integrals(vs, fpx[..., np.newaxis], line_forc[1:] = utilities.line_integrals(vs, fpx[..., np.newaxis], fpy[..., np.newaxis], kind='same')[1:] fpy[..., np.newaxis], kind='same')[1:] ... ...
 ... @@ -2,6 +2,9 @@ import xarray as xr ... @@ -2,6 +2,9 @@ import xarray as xr import matplotlib.pyplot as plt import matplotlib.pyplot as plt import numpy as np import numpy as np from IPython import get_ipython import matplotlib get_ipython().magic('matplotlib inline') #xr.open_dataset('zonjet.snapshot.nc').u.isel(x,Time=20).plot() #xr.open_dataset('zonjet.snapshot.nc').u.isel(x,Time=20).plot() ... @@ -48,7 +51,7 @@ class section: ... @@ -48,7 +51,7 @@ class section: pass pass def fileinfo(self): def fileinfo(self): print('Info: ', self.load().data_vars) print('Info: ', self.load()) try: try: ... @@ -56,9 +59,9 @@ try: ... @@ -56,9 +59,9 @@ try: except NameError: except NameError: pass pass p = section('zonjet.snapshot.nc','u') p = section('zonjet.averages.nc','rho') p.fileinfo() p.fileinfo() p.select(20,'zt',9).plot() p.select(0,'xt',13).plot.contourf(levels=11) #a.plot(vmax=0.5, vmin=-0.5)