43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
import xarray as xr
|
|
import glob
|
|
import numpy as np
|
|
def AddMMM(file="DHW_ssp245_BCC-CSM2-MR_DHW.nc",directoryData="in",directorySal="Sal"):
|
|
|
|
nc1 = xr.open_dataset("Corr_ALL_mmm.nc",decode_times=False,decode_timedelta=False)
|
|
nc2 = xr.open_dataset(directoryData+"/"+file,decode_times=False,decode_timedelta=False)
|
|
MapType=0
|
|
if file.count("BCC-CSM2-MR")>0:
|
|
MapType=0
|
|
if file.count("CanESM5")>0:
|
|
MapType=1
|
|
if file.count("CESM2")>0:
|
|
MapType=2
|
|
if file.count("EC-Earth3")>0:
|
|
MapType=3
|
|
if file.count("IPSL-CM6A-LR")>0:
|
|
MapType=4
|
|
if file.count("MIROC6")>0:
|
|
MapType=5
|
|
if file.count("MRI-ESM2-0")>0:
|
|
MapType=6
|
|
if file.count("NorESM2-MM")>0:
|
|
MapType=7
|
|
if file.count("Ensemble5")>0:
|
|
MapType=8
|
|
if file.count("Ensemble8")>0:
|
|
MapType=9
|
|
print(MapType)
|
|
ALL_mmm = 1 * np.ones((nc2.dims['lat'], nc2.dims['lon'])) * nc1.ALL_mmm.isel(time=MapType)
|
|
nc2["MMM"]=ALL_mmm
|
|
nc2.MMM.attrs={"long_name" : "monthly mean sea surface temperature climatology",
|
|
"standard_name" : "sea_surface_temperature",
|
|
"units":"degrees_Celsius"}
|
|
nc2.to_netcdf(path=directorySal+"/"+file)
|
|
|
|
|
|
for i in glob.glob("in/*"):
|
|
A=i.split("/")[-1]
|
|
SalidaDir="Sal"
|
|
print(A,i)
|
|
AddMMM(file=A,directoryData="in",directorySal="Sal")
|