Commit 2ca16923 authored by Patrick Wagner's avatar Patrick Wagner
Browse files

Initial commit

parents
Loading
Loading
Loading
Loading
+90 −0
Original line number Diff line number Diff line
%% Cell type:code id: tags:

``` 
%matplotlib inline
```

%% Cell type:code id: tags:

``` 
## Load required modules
import numpy as np
import matplotlib.pyplot as plt
import xarray as xr
```

%% Cell type:code id: tags:

``` 
```

%% Cell type:code id: tags:

``` 
## Load data
figdir="/gfs1/work/shkpwagn/ARIANE/VIKING20-K301_Turtle/FIGURES/"
outfile="/gfs1/work/shkpwagn/ARIANE/VIKING20-K301_Turtle/DATA/GB_arrival.nc"

ts=[]
ts_clim=[]

year1=1960
year2=1961#2007



for year in np.arange(year1,year2+1):
    dir="/gfs1/work/shkpwagn/ARIANE/VIKING20-K301_Turtle/DATA/GS-"+str(year)+"/ariane_trajectories_qualitative_GB_initt.nc"
    data=xr.open_dataset(dir)
    print(data)
    dummy=((year-1948)*365+(data['ts_gb']+data['init_t'])*5)
    ts=np.append(ts,dummy.values)
    dummy=(data['ts_gb']+data['init_t'])*5
    dummy[dummy>730]=dummy[dummy>730]-730
    dummy[dummy>365]=dummy[dummy>365]-365
    ts_clim=np.append(ts_clim,dummy.values)
    print(ts_clim.shape)


year=(ts/365).astype(int)+1948
```

%% Cell type:code id: tags:

``` 
## Plot

fig,ax=plt.subplots(2,1)
out="gb_arrival.pdf"

yebin=np.arange(year.min()-0.5,year.max()+1.5)
mdays=[0,31,28,31,30,31,30,31,31,30,31,30,31]
mlabel=['Jan','Feb','Mar','Arp','May','Jun','Jul','Aug','Sep','Oct','Nov','Dec',]
mobin=np.cumsum(mdays)

nyears, bins, patches=ax[0].hist(year,yebin)
nmonth, bins, patches=ax[1].hist(ts_clim,mobin)

cbin=bins[1:]-np.asarray(mdays[1:])/2
ax[1].set_xticks(cbin)
ax[1].set_xticklabels(mlabel, fontsize=12)

ax[0].set_ylabel('Floats')
ax[1].set_ylabel('Floats')

plt.savefig(figdir+out)
```

%% Output


%% Cell type:code id: tags:

``` 
## Save to file

dataout=xr.Dataset({'count': (['year'], nyears), 'count_clim': (['month'],nmonth)},
                   coords={'year': np.arange(year.min(),year.max()+1)})
dataout.to_netcdf(path=outfile)
```
+0 −0

File added.

Preview size limit exceeded, changes collapsed.

+0 −0

File added.

Preview size limit exceeded, changes collapsed.

+0 −0

File added.

Preview size limit exceeded, changes collapsed.

+0 −0

File added.

Preview size limit exceeded, changes collapsed.