Changeset 26180

Timestamp:

04/08/21 03:47:25 (4 years ago)

Author:

bdef

Message:

CHG: adding range type to some functions type possibilities

File:

• issm/trunk-jpl/src/m/miscellaneous/MatlabFuncs.py (modified) (1 diff)

issm/trunk-jpl/src/m/miscellaneous/MatlabFuncs.py

@@ -58 +58 @@
     import numpy as np
 
-    if not isinstance(s, (tuple, list, dict, np.ndarray)):
+    if not isinstance(s, (range, tuple, list, dict, np.ndarray)):
         s = [s]
 
-    if not isinstance(a, (tuple, list, dict, np.ndarray)):
+    if not isinstance(a, (range, tuple, list, dict, np.ndarray)):
         a = [a]
 
-    if not isinstance(a, np.ndarray):
+    if not isinstance(a, (range, np.ndarray)):
         b = [item in s for item in a]
 
     else:
-        if not isinstance(s, np.ndarray):
+        if not isinstance(s, (range, np.ndarray)):
             b = np.empty_like(a)
             for i, item in enumerate(a.flat):
                 b.flat[i] = item in s
+
         else:
             b = np.in1d(a.flat, s.flat).reshape(a.shape)