minor changes for aquastat field names

ec-jrc · Nov 15, 2024 · d85960e · d85960e
1 parent 74771e5
commit d85960e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/lisfloodutilities/water-demand-historic/step2_domestic_demand.py b/src/lisfloodutilities/water-demand-historic/step2_domestic_demand.py
@@ -174,7 +174,7 @@ def main():
         country_code = country_codes.iloc[jj]['country-code']
 
         # Find location of data in AQUASTAT table
-        sel = np.where((aquastat['m49']==country_code) & (aquastat['Variable Name']=='Municipal water withdrawal'))[0]
+        sel = np.where((aquastat['m49']==country_code) & (aquastat['Variable']=='Municipal water withdrawal'))[0]
         if len(sel)==0: 
             continue
 

diff --git a/src/lisfloodutilities/water-demand-historic/step3a_industrial_demand.py b/src/lisfloodutilities/water-demand-historic/step3a_industrial_demand.py
@@ -368,7 +368,7 @@ def main():
     for jj in np.arange(country_codes.shape[0]):
         country_code = country_codes.iloc[jj]['country-code']    
         for ii in np.arange(len(years)):
-            sel = (aquastat['m49']==country_code) & (aquastat['Variable Name']=='Industrial water withdrawal') & (aquastat['Year']==years[ii])
+            sel = (aquastat['m49']==country_code) & (aquastat['Variable']=='Industrial water withdrawal') & (aquastat['Year']==years[ii])
             if np.sum(sel)==0: continue
             table_aquastat_industry_withdrawal[jj,ii] = aquastat['Value'][sel].values # km3/year