scahpy 0.0.10__tar.gz → 0.0.11__tar.gz

{scahpy-0.0.10/scahpy.egg-info → scahpy-0.0.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: scahpy
-Version: 0.0.10
+Version: 0.0.11
 Summary: Package to process and analyze outputs from IGP-RESM-COW model
 Author-email: Fiorela Castillón <fv.castillon@gmail.com>
 License:                     GNU GENERAL PUBLIC LICENSE
@@ -689,6 +689,7 @@ License-File: LICENSE
 Requires-Dist: wrf-python
 Requires-Dist: netCDF4
 Requires-Dist: xarray
+Requires-Dist: dask
 Requires-Dist: numpy
 Requires-Dist: pandas
 Requires-Dist: matplotlib

{scahpy-0.0.10 → scahpy-0.0.11}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "scahpy"
-version = "0.0.10"
+version = "0.0.11"
 description = "Package to process and analyze outputs from IGP-RESM-COW model"
 readme = "README.md"
 authors = [{ name = "Fiorela Castillón", email = "fv.castillon@gmail.com" }]
@@ -22,6 +22,7 @@ dependencies = [
         'wrf-python',
         'netCDF4',
         'xarray',
+        'dask',
         'numpy',
         'pandas',
         'matplotlib',      

{scahpy-0.0.10 → scahpy-0.0.11}/scahpy/__init__.py

@@ -9,4 +9,4 @@ __all__ = [
 ]
 
 # Version of scahpy package
-__version__ = "0.0.10"
+__version__ = "0.0.11"

{scahpy-0.0.10 → scahpy-0.0.11}/scahpy/in_out.py

@@ -127,50 +127,90 @@ def _select_time(x,difHor,sign):
     d=d.assign_coords({'time':time2})
     return d
 
-def ds_wrf_multi(files,list_no_vars,difHor=0,sign=1):
-    """Read a list of wrfout files for the variables selected.
-    ES: Lee una lista de archivos wrfout para las variables seleccionadas.
+def ds_wrf_multi(files, list_no_vars, difHor=0, sign=1, save_path=None):
+    """
+    Read a list of wrfout files for the selected variables and optionally save the resulting netCDF file.
 
-    Parameters/Parametros:
-    ----------------------
-    files : List of wrfout files / Lista de archivos wrfout
-    list_no_vars : List of variables to be delated / Lista de variables a ignorar
-    difHor : String with the hours t / Lista de variables a mantener
-    sign: -1 or 1 according to the difference / +1 o -1 dependiendo de
-    la diferencia horaria respecto a la UTC
+    Parameters:
+    -----------
+    files : list of str
+        List of paths to wrfout files.
+    list_no_vars : list
+        List of variables to be excluded.
+    difHor : str, optional
+        String with the hour difference.
+    sign : int, optional
+        -1 or 1 according to the difference.
+    save_path : str, optional
+        Path to save the resulting netCDF file.
+
+    Returns:
+    --------
+    ds1 : xarray.Dataset
+        Dataset containing the selected variables.
     """
-    ds = xr.open_mfdataset(files, combine='nested', concat_dim='time', parallel= True ,engine='netcdf4',
-                                drop_variables=list_no_vars, preprocess = partial(_select_time,difHor=difHor,sign=sign))
-    ds.attrs = []
+    # Read the wrfout files and drop specified variables
+    ds = xr.open_mfdataset(files, combine='nested', concat_dim='time', parallel=True, engine='netcdf4',
+                            drop_variables=list_no_vars, preprocess=partial(_select_time, difHor=difHor, sign=sign))
+
+    # Remove duplicate times
     _, index = np.unique(ds['time'], return_index=True)
     ds = ds.isel(time=index)
-    ds1 = _new_coords(files[0],ds)
-    ds1.encoding['unlimited_dims']=('time',)
-   
-    return ds1 
 
-def ds_wrf_single(file,list_no_vars,difHor=0,sign=1):
-    """Read a list of wrfout files for the variables selected.
-    ES: Lee una lista de archivos wrfout para las variables seleccionadas.
+    # Update coordinates and encoding
+    ds1 = _new_coords(files[0], ds)
+    ds1.encoding['unlimited_dims'] = ('time',)
 
-    Parameters/Parametros:
-    ----------------------
-    files : wrfout file / archivo wrfout
-    list_no_vars : List of variables to be delated / Lista de variables a ignorar
-    difHor : String with the hours t / Lista de variables a mantener
-    sign: -1 or 1 according to the difference / +1 o -1 dependiendo de
-    la diferencia horaria respecto a la UTC
+    # Optionally save the resulting netCDF file
+    if save_path is not None:
+        ds1.to_netcdf(save_path)
+
+    return ds1
+
+
+def ds_wrf_single(file, list_no_vars, difHor=0, sign=1, save_path=None):
+    """
+    Read a list of wrfout files for the selected variables and optionally save the resulting netCDF file.
+
+    Parameters:
+    -----------
+    file : str 
+        Path to the wrfout file.
+    list_no_vars : list
+        List of variables to be excluded.
+    difHor : str, optional
+        String with the hour difference.
+    sign : int, optional
+        -1 or 1 according to the difference.
+    save_path : str, optional
+        Path to save the resulting netCDF file.
+
+    Returns:
+    --------
+    ds2 : xarray.Dataset
+        Dataset containing the selected variables.
     """
-    ds = xr.open_dataset(file,engine='netcdf4', drop_variables=list_no_vars)
-    ds1 = ds.rename({'XTIME':'time'}).swap_dims({'Time':'time'})
-    time2=pd.to_datetime(ds1.time.values) + (sign*pd.Timedelta(difHor))
-    ds1=ds1.assign_coords({'time':time2})
+    # Read the wrfout file(s) and drop specified variables
+    ds = xr.open_dataset(file, engine='netcdf4', drop_variables=list_no_vars)
+
+    # Rename and manipulate time coordinate
+    ds1 = ds.rename({'XTIME': 'time'}).swap_dims({'Time': 'time'})
+    time2 = pd.to_datetime(ds1.time.values) + (sign * pd.Timedelta(difHor))
+    ds1 = ds1.assign_coords({'time': time2})
     ds1.attrs = []
+
+    # Remove duplicate times
     _, index = np.unique(ds1['time'], return_index=True)
     ds1 = ds1.isel(time=index)
-    ds2 = _new_coords(file,ds1)
-    ds2.encoding['unlimited_dims']=('time',)
-   
+
+    # Update coordinates and encoding
+    ds2 = _new_coords(file, ds1)
+    ds2.encoding['unlimited_dims'] = ('time',)
+
+    # Optionally save the resulting netCDF file
+    if save_path is not None:
+        ds2.to_netcdf(save_path)
+
     return ds2
 
 

scahpy-0.0.11/scahpy/map_plots.py

@@ -0,0 +1,279 @@
+import matplotlib.pyplot as plt
+import numpy as np
+import cartopy.crs as ccrs
+import cartopy.feature as cfe
+from cartopy.mpl.ticker import LongitudeFormatter, LatitudeFormatter
+from cartopy.io.shapereader import Reader
+import cmocean
+import matplotlib.colors
+import importlib.resources
+from pathlib import Path
+
+def map_pp_uv10_sst(pp, sfc, levs, levs2, shapefile=None, output_path=None, save_maps=True, freq=None,
+                            quiverkey_speed=5,extent=None):
+    """
+    Plots precipitation maps for specified months.
+
+    Parameters:
+    - pp (xarray.Dataset): Precipitation data.
+    - sfc (xarray.Dataset): Surface data with SST and UV10 data.
+    - levs (list): Contour levels for precipitation.
+    - levs2 (list): Contour levels for SSTSK.
+    - shapefile (str): Path to the shapefile for South America countries (could be any other).
+    - output_path (str, optional): Path to save the maps. If None, maps will be displayed but not saved. Defaults to None.
+    - freq (str): Could be 'H' , 'D', 'M', 'Y'. Defaults to Hourly ('H').
+    - save_maps (bool, optional): If True, saves the maps. If False, only displays them. Defaults to True.
+    - quiverkey_speed (int, optional): Speed parameter for quiverkey. Defaults to 5.
+    - extent: [x1,x2,y1,y2] spatial extension
+    """
+    if shapefile is None:
+        data_dir = importlib.resources.files('scahpy.data')
+        data_path = Path(data_dir , 'SA_paises.shp')
+        sa = cfe.ShapelyFeature(Reader(data_path).geometries(), ccrs.PlateCarree(), edgecolor='k', facecolor='none')
+    else:
+        sa = cfe.ShapelyFeature(Reader(shapefile).geometries(), ccrs.PlateCarree(), edgecolor='k', facecolor='none')
+
+    if extent is None:
+        extent = [-93.8, -68, -21.9, 4.1]
+    
+    cmaps = cmocean.tools.lighten(cmocean.cm.rain, 0.85)
+    norm = matplotlib.colors.BoundaryNorm(levs, cmaps.N)
+
+    lons = pp.lon.values
+    lats = pp.lat.values
+    times = np.datetime_as_string(sfc.time.values,unit='m')
+
+    for i in range(len(times)):
+        if freq == 'H' or freq is None:
+            timess = times[i][0:13]
+        elif freq == 'D':
+            timess = times[i][0:10]
+        elif freq == 'M':
+            timess = times[i][0:7]
+        elif freq == 'Y':
+            timess = times[i][0:4]
+
+        fig, axs = plt.subplots(figsize=(13, 12), ncols=1, nrows=1, sharex=True, sharey=True,
+                                subplot_kw=dict(projection=ccrs.PlateCarree()))
+
+        pcm = axs.contourf(lons, lats, pp.sel(time=timess), levels=levs,
+                            cmap=cmaps, norm=norm, extend='both', transform=ccrs.PlateCarree())
+        fig.colorbar(pcm, ax=axs, label='mm', orientation='vertical', shrink=.7, pad=0.07, aspect=20, format='%3.0f')
+        c = axs.contour(lons, lats, sfc.get('SSTSK').sel(time=timess),
+                        levels=levs2, colors=['#F29727', '#C70039', '#511F73'],
+                        linewidths=[1.5, 1.6, 1.8], linestyles='solid',
+                        alpha=0.45, transform=ccrs.PlateCarree(), zorder=7)
+        axs.clabel(c, levels=levs2, inline=False, colors='#000000', fontsize=12, zorder=9)
+        Q = axs.quiver(lons[::7], lats[::7],
+                        sfc.U10.sel(time=timess)[::7, ::7], sfc.V10.sel(time=timess)[::7, ::7],
+                        headwidth=5, headlength=7)
+        axs.quiverkey(Q, 0.87, 1.02, quiverkey_speed, f'{quiverkey_speed} m/s', labelpos='E', coordinates='axes', labelsep=0.05)
+
+        axs.add_feature(sa, linewidth=0.7, zorder=7)
+        lon_formatter = LongitudeFormatter()
+        lat_formatter = LatitudeFormatter()
+        axs.yaxis.set_major_formatter(lat_formatter)
+        axs.xaxis.set_major_formatter(lon_formatter)
+        axs.set_extent(extent)
+
+        plt.title(f'Map of Precipitation : {timess}')
+
+        if save_maps:
+            if output_path is None:
+                raise ValueError("Output path cannot be None when saving maps.")
+            plt.savefig(f'{output_path}/m_TSM_UV_PP_{timess}.png',
+                        bbox_inches='tight', dpi=300, facecolor='white', transparent=False)
+        else:
+            plt.show()
+
+        plt.close()
+
+def map_clim_pp_uv10_sst(pp, sfc, levs, levs2, shapefile, output_path=None, month_list=None,
+                           save_maps=True, quiverkey_speed=5,extent=None):
+    """
+    Plots precipitation maps for specified months.
+
+    Parameters:
+    - pp (xarray.Dataset): Precipitation data.
+    - sfc (xarray.Dataset): Surface data with SST and UV10 data.
+    - levs (list): Contour levels for precipitation.
+    - levs2 (list): Contour levels for SSTSK.
+    - shapefile (str): Path to the shapefile for South America countries (could be any other).
+    - output_path (str, optional): Path to save the maps. If None, maps will be displayed but not saved. Defaults to None.
+    - month_list (list, optional): List of months to plot. Defaults to None (all months).
+    - save_maps (bool, optional): If True, saves the maps. If False, only displays them. Defaults to True.
+    - quiverkey_speed (int, optional): Speed parameter for quiverkey. Defaults to 5.
+    - extent: [x1,x2,y1,y2] spatial extension
+    """
+    if shapefile is None:
+        data_dir = importlib.resources.files('scahpy.data')
+        data_path = Path(data_dir , 'SA_paises.shp')
+        sa = cfe.ShapelyFeature(Reader(data_path).geometries(), ccrs.PlateCarree(), edgecolor='k', facecolor='none')
+    else:
+        sa = cfe.ShapelyFeature(Reader(shapefile).geometries(), ccrs.PlateCarree(), edgecolor='k', facecolor='none')
+
+    if extent is None:
+        extent = [-93.8, -68, -21.9, 4.1]
+
+    cmaps = cmocean.tools.lighten(cmocean.cm.rain, 0.85)
+    norm = matplotlib.colors.BoundaryNorm(levs, cmaps.N)
+
+    lons = pp.lon.values
+    lats = pp.lat.values
+
+
+
+    if month_list is None:
+        month_list = range(1, 13)
+
+    for month in month_list:
+        fig, axs = plt.subplots(figsize=(13, 12), ncols=1, nrows=1, sharex=True, sharey=True,
+                                subplot_kw=dict(projection=ccrs.PlateCarree()))
+
+        pcm = axs.contourf(lons, lats, pp.sel(month=month), levels=levs,
+                            cmap=cmaps, norm=norm, extend='both', transform=ccrs.PlateCarree())
+        fig.colorbar(pcm, ax=axs, label='mm/month', orientation='vertical', shrink=.7, pad=0.07, aspect=20, format='%3.0f')
+        c = axs.contour(lons, lats, sfc.get('SSTSK').sel(month=month),
+                        levels=levs2, colors=['#F29727', '#C70039', '#511F73'],
+                        linewidths=[1.5, 1.6, 1.8], linestyles='solid',
+                        alpha=0.45, transform=ccrs.PlateCarree(), zorder=7)
+        axs.clabel(c, levels=levs2, inline=False, colors='#000000', fontsize=12, zorder=9)
+        Q = axs.quiver(lons[::7], lats[::7],
+                        sfc.U10.sel(month=month)[::7, ::7], sfc.V10.sel(month=month)[::7, ::7],
+                        headwidth=5, headlength=7)
+        axs.quiverkey(Q, 0.87, 1.02, quiverkey_speed, f'{quiverkey_speed} m/s', labelpos='E', coordinates='axes', labelsep=0.05)
+
+        axs.add_feature(sa, linewidth=0.7, zorder=7)
+        lon_formatter = LongitudeFormatter()
+        lat_formatter = LatitudeFormatter()
+        axs.yaxis.set_major_formatter(lat_formatter)
+        axs.xaxis.set_major_formatter(lon_formatter)
+        axs.set_extent(extent)
+
+        plt.title(f'Map of Precipitation Month: {month:02d}')
+
+        if save_maps:
+            if output_path is None:
+                raise ValueError("Output path cannot be None when saving maps.")
+            plt.savefig(f'{output_path}/m_TSM_UV_PP_{month}.png',
+                        bbox_inches='tight', dpi=300, facecolor='white', transparent=False)
+        else:
+            plt.show()
+
+        plt.close()
+
+def cross_section_yz(ds,vari,levs,cmaps,title,quiverkey_speed=5,
+                     output_path=None, freq=None, save_maps=True):
+    """
+    Plots Cross Sections of Pressure level variables in YZ direction.
+
+    Parameters:
+    - ds (xarray.Dataset): wrfout data already interpolated to pressure levels (output from spatial_scales.vert_levs()).
+    - vari (list): List of variables to plot including V and W.
+    - levs (list): Contour levels for variable.
+    - title (str): Title to display in the graph.
+    - output_path (str, optional): Path to save the maps. If None, maps will be displayed but not saved. Defaults to None.
+    - freq (str): Could be 'H' , 'D', 'M', 'Y'. Defaults to Hourly ('H').
+    - save_maps (bool, optional): If True, saves the maps. If False, only displays them. Defaults to True.
+    - quiverkey_speed (int, optional): Speed parameter for quiverkey. Defaults to 5.
+    """
+
+    lats = ds.lat.values
+    times = np.datetime_as_string(ds.time.values,unit='m')
+    Y=ds.level.values
+    
+    norms = matplotlib.colors.BoundaryNorm(levs, cmaps.N)
+    plt.rcParams['font.family'] = 'Monospace'
+
+    if freq == 'H' or freq is None:
+        j=13
+    elif freq == 'D':
+        j=10
+    elif freq == 'M':
+        j=7
+    elif freq == 'Y':
+        j=4   
+    
+    for i in range(len(times)):
+            
+        timess = times[i][0:j]
+        
+        vv=ds.get('V').sel(time=timess)
+        ww=ds.get('W').sel(time=timess)*1000
+        var=ds.get(vari).sel(time=timess) #(g/kg)
+        
+        fig,axs = plt.subplots(figsize=(6,10),ncols=1,nrows=1)
+        pcm=axs.contourf(lats, Y, var,levels=levs,cmap=cmaps,norm=norms,extend='both')
+        fig.colorbar(pcm,ax=axs,label='', orientation='horizontal', shrink=.9,pad=0.04,aspect=25,format='%3.1f')
+        Q=axs.quiver(lats[::4],Y,vv[::,::4],ww[::,::4],scale=170,headwidth=4,headlength=4)
+        axs.invert_yaxis()
+        axs.quiverkey(Q,0.87,0.98,quiverkey_speed, f'{quiverkey_speed} m/s',labelpos='E',coordinates='axes',labelsep=0.05) 
+        plt.title(f'{title} {timess}',loc='center')
+    
+        if save_maps:
+            if output_path is None:
+                raise ValueError("Output path cannot be None when saving maps.")
+            plt.savefig(f'{output_path}/Cross_Section_YZ_{var}_{timess}.png',
+                        bbox_inches='tight', dpi=300, facecolor='white', transparent=False)
+        else:
+            plt.show()
+
+        plt.close()
+
+def cross_section_xz(ds,vari,levs,cmaps,title,quiverkey_speed=5,
+                     output_path=None, freq=None, save_maps=True):
+    """
+    Plots Cross Sections of Pressure level variables in XZ direction.
+
+    Parameters:
+    - ds (xarray.Dataset): wrfout data already interpolated to pressure levels (output from spatial_scales.vert_levs()).
+    - vari (list): List of variables to plot including V and W.
+    - levs (list): Contour levels for variable.
+    - title (str): Title to display in the graph.
+    - output_path (str, optional): Path to save the maps. If None, maps will be displayed but not saved. Defaults to None.
+    - freq (str): Could be 'H' , 'D', 'M', 'Y'. Defaults to Hourly ('H').
+    - save_maps (bool, optional): If True, saves the maps. If False, only displays them. Defaults to True.
+    - quiverkey_speed (int, optional): Speed parameter for quiverkey. Defaults to 5.
+    """
+
+    lons = ds.lon.values
+    times = np.datetime_as_string(ds.time.values,unit='m')
+    Y=ds.level.values
+    
+    norms = matplotlib.colors.BoundaryNorm(levs, cmaps.N)
+    plt.rcParams['font.family'] = 'Monospace'
+
+    if freq == 'H' or freq is None:
+        j=13
+    elif freq == 'D':
+        j=10
+    elif freq == 'M':
+        j=7
+    elif freq == 'Y':
+        j=4   
+    
+    for i in range(len(times)):
+            
+        timess = times[i][0:j]
+        
+        uu=ds.get('U').sel(time=timess)
+        ww=ds.get('W').sel(time=timess)*1000
+        var=ds.get(vari).sel(time=timess) #(g/kg)
+        
+        fig,axs = plt.subplots(figsize=(6,10),ncols=1,nrows=1)
+        pcm=axs.contourf(lons, Y, var,levels=levs,cmap=cmaps,norm=norms,extend='both')
+        fig.colorbar(pcm,ax=axs,label='', orientation='horizontal', shrink=.9,pad=0.04,aspect=25,format='%3.1f')
+        Q=axs.quiver(lons[::4],Y,uu[::,::4],ww[::,::4],scale=170,headwidth=4,headlength=4)
+        axs.invert_yaxis()
+        axs.quiverkey(Q,0.87,0.98,quiverkey_speed, f'{quiverkey_speed} m/s',labelpos='E',coordinates='axes',labelsep=0.05) 
+        plt.title(f'{title} {timess}',loc='center')
+    
+        if save_maps:
+            if output_path is None:
+                raise ValueError("Output path cannot be None when saving maps.")
+            plt.savefig(f'{output_path}/Cross_Section_XZ_{var}_{timess}.png',
+                        bbox_inches='tight', dpi=300, facecolor='white', transparent=False)
+        else:
+            plt.show()
+
+        plt.close()

{scahpy-0.0.10 → scahpy-0.0.11}/scahpy/met_diag.py

@@ -1,29 +1,47 @@
 import numpy as np
 import xarray as xr
 
-def calc_pp(ds,elim=False):
-    """ de-acumulate the rainfall and save it as PP.
-    ES: Calcula la precipitación nominal en el tiempo de salida (ej. 3hr, etc),
-    es decir desacumula la precipitación líquida y la guarda como 'PP'.
-
-    Parameters/Parámetros:
-    ----------------------
-    ds : dataset with the variables RAINC, RAINNC and RAINSH already loaded with 
-    coordinates already processed / dataset con las variables RAINC, RAINNC and RAINSH 
-    ya cargado con las coordenadas apropiadas.
+def calc_pp(ds, vars_to_sum=['RAINC', 'RAINNC', 'RAINSH'], elim=False):
+    """Calculate precipitation nominal at the output time (e.g., 3hr, etc).
+    De-accumulate liquid precipitation and save it as 'PP'.
+
+    Parameters:
+    -----------
+    ds : xarray.Dataset
+        Dataset with the variables RAINC, RAINNC, and RAINSH already loaded and processed coordinates.
+    
+    vars_to_sum : list of str, optional
+        List of variables to be summed for precipitation calculation (default is ['RAINC', 'RAINNC', 'RAINSH']).
+
+    elim : bool, optional
+        If True, eliminates intermediate variables after calculation (default is False).
+
+    Returns:
+    --------
+    ds : xarray.Dataset
+        Dataset with calculated precipitation 'PP'.
+
+    Example:
+    ---------
+    # Calculate precipitation with default settings (sum of RAINC, RAINNC, and RAINSH)
+    ds = calc_pp(ds)
+
+    # Calculate precipitation using only RAINC and RAINNC
+    ds = calc_pp(ds, vars_to_sum=['RAINC', 'RAINNC'])
     """
     ntime = ds.time[0:-1]
-    ds['PP2'] = ds['RAINC'] + ds['RAINNC'] + ds['RAINSH']
+    # Calculate precipitation based on the sum of specified variables
+    ds['PP2'] = sum(ds[var] for var in vars_to_sum)
 
-    dd=ds['PP2'].diff('time')
+    # De-accumulate precipitation and save it as 'PP'
+    dd = ds['PP2'].diff('time')
     dd['time'] = ntime
 
-    ds['PP'] = dd
-
-    if elim==True:
-        ds=ds.drop_vars(['PP2','RAINC','RAINNC','RAINSH'])
+    # Drop intermediate variables if elim is True
+    if elim:
+        ds = ds.drop_vars(['PP2'] + vars_to_sum)
     else:
-        ds=ds.drop_vars(['PP2'])
+        ds = ds.drop_vars(['PP2'])
 
     return ds
 
@@ -98,4 +116,19 @@ def calc_qe(ds,elim=False):
         ds=ds.drop_vars(['QVAPOR'])
 
     return ds
+
+def calc_celsius(ds,var):
+    """ Calculate de conversion from Kelvin to Celsius'
+
+                                K-273.15 = C
+
+    Parameters/Parámetros:
+    ----------------------
+    ds : dataset with the SSTSK variable/ dataset con las variables U10 and V10 
+    ya cargado con las coordenadas apropiadas.
+    """
+
+    ds[var]=ds[var]-273.15
+
+    return ds
  

{scahpy-0.0.10 → scahpy-0.0.11}/scahpy/spatial_scales.py

@@ -29,7 +29,7 @@ def vert_levs(ds,varis,lvls=None):
         dlvl[var].attrs['vert_units'] = ''
         dlvl[var].encoding['coordinates'] = 'time lat lon'
         datasets.append(dlvl)
-    ds_lvl = xr.concat(datasets,dim='time')
+    ds_lvl = xr.merge(datasets)
     ds_lvl.encoding['unlimited_dims']=('time',)
     ds_lvl['lat'].attrs = {"units": 'degrees_north', 'axis': 'Y','long_name':'Latitude','standard_name':'latitude'}
     ds_lvl['lon'].attrs = {"units": 'degrees_east', 'axis': 'X','long_name':'Longitude','standard_name':'longitude'}

{scahpy-0.0.10 → scahpy-0.0.11}/scahpy/temp_scales.py

@@ -1,7 +1,7 @@
 import xarray as xr
 import numpy as np
 
-def dmy_var(ds,tiempo=None ,accum=None, avg=None, mediana=None):
+def dmy_var(ds, tiempo=None, accum=None, avg=None, mediana=None):
     """Convert hourly (default wrf out) time to any acceptable by resample function.
     ES: Convierte los datos horarios (por defecto wrfout) a otra escala de tiempo 
     aceptada por la función resample (ejm. 'D' diario, 'M' mensual, etc)
@@ -15,19 +15,30 @@ def dmy_var(ds,tiempo=None ,accum=None, avg=None, mediana=None):
     mediana : if True use the median function / Si es True, emplea la función mediana
     """
 
+    datasets = []
+
     if accum is not None:
-        ds_ac = ds[accum].resample(time = tiempo).sum()
+        ds_ac = ds[accum].resample(time=tiempo).sum()
+        datasets.append(ds_ac)
+
     if avg is not None:
-        ds_avg = ds[avg].resample(time = tiempo).mean()
+        ds_avg = ds[avg].resample(time=tiempo).mean()
+        datasets.append(ds_avg)
+
     if mediana is not None:
-        ds_med = ds[mediana].resample(time = tiempo).median()
+        ds_med = ds[mediana].resample(time=tiempo).median()
+        datasets.append(ds_med)
+
+    if not datasets:  # If no datasets were created
+        print('Ingrese al menos una variable para operar.')
+        return None
 
     try:
-        datasets = [ds_ac, ds_avg, ds_med]
         ds_all = xr.concat(datasets, dim='time')
-    except:
+    except ValueError:
+        print('Coloque una escala temporal apropiada')
+        return None
 
-        print('coloque una escala temporal apropiada')
     return ds_all
 
 

{scahpy-0.0.10 → scahpy-0.0.11/scahpy.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: scahpy
-Version: 0.0.10
+Version: 0.0.11
 Summary: Package to process and analyze outputs from IGP-RESM-COW model
 Author-email: Fiorela Castillón <fv.castillon@gmail.com>
 License:                     GNU GENERAL PUBLIC LICENSE
@@ -689,6 +689,7 @@ License-File: LICENSE
 Requires-Dist: wrf-python
 Requires-Dist: netCDF4
 Requires-Dist: xarray
+Requires-Dist: dask
 Requires-Dist: numpy
 Requires-Dist: pandas
 Requires-Dist: matplotlib

{scahpy-0.0.10 → scahpy-0.0.11}/scahpy.egg-info/requires.txt

@@ -1,6 +1,7 @@
 wrf-python
 netCDF4
 xarray
+dask
 numpy
 pandas
 matplotlib

scahpy-0.0.10/scahpy/map_plots.py

@@ -1,73 +0,0 @@
-import matplotlib.pyplot as plt
-import numpy as np
-import cartopy.crs as ccrs
-import cartopy.feature as cfe
-from cartopy.mpl.ticker import LongitudeFormatter, LatitudeFormatter
-from cartopy.io.shapereader import Reader
-import cmocean
-import matplotlib.colors
-
-def m_pp_uv10_sst(pp, sfc, levs, levs2, sa_shapefile, output_path=None, month_list=None,
-                           save_maps=True, quiverkey_speed=5,extent=None):
-    """
-    Plots precipitation maps for specified months.
-
-    Parameters:
-    - pp (xarray.Dataset): Precipitation data.
-    - sfc (xarray.Dataset): Surface data with SST and UV10 data.
-    - levs (list): Contour levels for precipitation.
-    - levs2 (list): Contour levels for SSTSK.
-    - sa_shapefile (str): Path to the shapefile for South America countries (could be any other).
-    - output_path (str, optional): Path to save the maps. If None, maps will be displayed but not saved. Defaults to None.
-    - month_list (list, optional): List of months to plot. Defaults to None (all months).
-    - save_maps (bool, optional): If True, saves the maps. If False, only displays them. Defaults to True.
-    - quiverkey_speed (int, optional): Speed parameter for quiverkey. Defaults to 5.
-    - extent: [x1,x2,y1,y2] spatial extension
-    """
-    cmaps = cmocean.tools.lighten(cmocean.cm.rain, 0.85)
-    norm = matplotlib.colors.BoundaryNorm(levs, cmaps.N)
-
-    lons = pp.lon.values
-    lats = pp.lat.values
-
-    sa = cfe.ShapelyFeature(Reader(sa_shapefile).geometries(), ccrs.PlateCarree(), edgecolor='k', facecolor='none')
-
-    if month_list is None:
-        month_list = range(1, 13)
-
-    for month in month_list:
-        fig, axs = plt.subplots(figsize=(13, 12), ncols=1, nrows=1, sharex=True, sharey=True,
-                                subplot_kw=dict(projection=ccrs.PlateCarree()))
-
-        pcm = axs.contourf(lons, lats, pp.sel(month=month), levels=levs,
-                            cmap=cmaps, norm=norm, extend='both', transform=ccrs.PlateCarree())
-        fig.colorbar(pcm, ax=axs, label='mm/month', orientation='vertical', shrink=.7, pad=0.07, aspect=20, format='%3.0f')
-        c = axs.contour(lons, lats, sfc.get('SSTSK').sel(month=month),
-                        levels=levs2, colors=['#F29727', '#C70039', '#511F73'],
-                        linewidths=[1.5, 1.6, 1.8], linestyles='solid',
-                        alpha=0.45, transform=ccrs.PlateCarree(), zorder=7)
-        axs.clabel(c, levels=levs2, inline=False, colors='#000000', fontsize=12, zorder=9)
-        Q = axs.quiver(lons[::7], lats[::7],
-                        sfc.U10.sel(month=month)[::7, ::7], sfc.V10.sel(month=month)[::7, ::7],
-                        headwidth=5, headlength=7)
-        axs.quiverkey(Q, 0.87, 1.02, quiverkey_speed, f'{quiverkey_speed} m/s', labelpos='E', coordinates='axes', labelsep=0.05)
-
-        axs.add_feature(sa, linewidth=0.7, zorder=7)
-        lon_formatter = LongitudeFormatter()
-        lat_formatter = LatitudeFormatter()
-        axs.yaxis.set_major_formatter(lat_formatter)
-        axs.xaxis.set_major_formatter(lon_formatter)
-        axs.set_extent(extent)
-
-        plt.title(f'Map of Precipitation Month: {month:02d}')
-
-        if save_maps:
-            if output_path is None:
-                raise ValueError("Output path cannot be None when saving maps.")
-            plt.savefig(f'{output_path}/m_TSM_UV_PP_{month}.png',
-                        bbox_inches='tight', dpi=300, facecolor='white', transparent=False)
-        else:
-            plt.show()
-
-        plt.close()
-