lookup + faster not

unbornchikken · unbornchikken · commit 10757d1ddda2 · 2017-01-16T21:51:14.000+01:00
diff --git a/arrayfire/array.py b/arrayfire/array.py
@@ -1008,7 +1008,9 @@ def __invert__(self):
         """
         Return ~self
         """
-        return self == 0
+        out = Array()
+        safe_call(backend.get().af_not(c_pointer(out.arr), self.arr))
+        self = out
 
     def __nonzero__(self):
         return self != 0
diff --git a/arrayfire/data.py b/arrayfire/data.py
@@ -799,3 +799,46 @@ def replace(lhs, cond, rhs):
         safe_call(backend.get().af_replace(lhs.arr, cond.arr, rhs.arr))
     else:
         safe_call(backend.get().af_replace_scalar(lhs.arr, cond.arr, c_double_t(rhs)))
+
+def lookup(a, idx, dim=0):
+    """
+    Lookup the values of input array based on index.
+
+    Parameters
+    ----------
+
+    a : af.Array.
+       Multi dimensional array.
+
+    idx : is lookup indices
+
+    dim : optional: int. default: 0.
+       Specifies the dimension for indexing
+
+    Returns
+    -------
+
+    out : af.Array
+          An array containing values at locations specified by 'idx'
+
+    Examples
+    ---------
+
+    >>> import arrayfire as af
+    >>> a = af.randu(3, 3)
+    >>> af.display(a)
+    [3 3 1 1]
+        0.7269     0.3569     0.3341
+        0.7104     0.1437     0.0899
+        0.5201     0.4563     0.5363
+
+    >>> idx = af.array([1, 2])
+    >>> o = af.lookup()
+    [2 1 1 1]
+        0.7269     0.3569     0.3341
+        0.7104     0.1437     0.0899
+
+    """
+    out = Array()
+    safe_call(backend.get().af_lookup(c_pointer(out.arr), a.arr, idx.arr, c_int_t(dim)))
+    return out
