Reproduce matplotlib.mlab.find

Apply kthyng#12 (comment)
douglasnehme · Sep 16, 2021 · 97ababc · 97ababc
1 parent 6cc5300
commit 97ababc
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tracpy/tools.py b/tracpy/tools.py
@@ -9,6 +9,7 @@
 * convert_indices
 * check_points
 * seed
+* find
 """
 
 import numpy as np
@@ -476,3 +477,13 @@ def make_proj(setup='nwgom', usebasemap=True, **kwargs):
         proj = Proj(**inputs)
 
     return proj
+
+
+def find(condition):
+    """
+    https://github.com/kthyng/tracpy/issues/12#issuecomment-555493595
+    """
+
+    res, = np.nonzero(np.ravel(condition))
+
+    return res