Skip to content

Instantly share code, notes, and snippets.

@spestana
Created October 23, 2018 19:56
Show Gist options
  • Save spestana/841468ea0770c9af0a3dfd2127f591ec to your computer and use it in GitHub Desktop.
Save spestana/841468ea0770c9af0a3dfd2127f591ec to your computer and use it in GitHub Desktop.
Create dataframe from SUMMA output netCDF files for SWE and snow water drainage
def create_dfs(path,file):
'''create df for swe, snow drainage from SUMMA .nc file'''
ds = xr.open_dataset(path+file)
# create swe dataframe
df_swe = ds.scalarSWE.to_dataframe()
df_swe = df_swe.reset_index(level='hru')
df_swe.drop(['hru'], 1, inplace=True)
# create snow drainage dataframe
df_sd = ds.scalarSnowDrainage.to_dataframe()
df_sd = df_sd.reset_index(level='hru')
df_sd.drop(['hru'], 1, inplace=True)
return [df_swe, df_sd]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment