Changeset 26818

Timestamp:

01/25/22 13:26:22 (3 years ago)

Author:

vverjans

Message:

BUG: working on Python syntax wrt revision 26810

File:

• issm/trunk-jpl/src/m/classes/SMBautoregression.py (modified) (2 diffs)

issm/trunk-jpl/src/m/classes/SMBautoregression.py ¶

@@ -129 +129 @@
             temprefelevation = np.zeros((md.smb.num_basins))
             areas = GetAreas(md.mesh.elements, md.mesh.x, md.mesh.y)
-            for ii in range(int(md.smb.num_basins)):
-                indices = np.where(md.smb.basin_id==ii+1)[0]
+            for ii,bid in enumerate(np.unique(md.smb.basin_id)):
+                indices = np.where(md.smb.basin_id==bid)[0]
                 elemsh  = np.zeros((len(indices)))
                 for jj in range(len(indices)):
@@ -142 +142 @@
                     elemsh[jj] = sumval/cntval
                 temprefelevation[ii] = np.sum(areas[indices]*elemsh)/np.sum(areas[indices])
-                print(indices)
-                print(areas[indices])
-                print(elemsh)
             if(np.any(templapserate_pos!=0) or np.any(templapserate_neg!=0)):
                 print('      smb.refelevation not specified: Reference elevations set to mean surface elevation of basins')