[python-arrayfire] 26/250: Adding statistics functions and relevant tests

[Ghislain Vaillant]

This is an automated email from the git hooks/post-receive script.

ghisvail-guest pushed a commit to branch debian/master
in repository python-arrayfire.

commit 96c5f9e1ac41b2b17c4c3816cda398c62123e0c4
Author: Pavan Yalamanchili <pavan at arrayfire.com>
Date:   Sat Jul 4 18:22:55 2015 -0400

    Adding statistics functions and relevant tests
---
 arrayfire/__init__.py      |  1 +
 arrayfire/statistics.py    | 97 ++++++++++++++++++++++++++++++++++++++++++++++
 tests/simple_statistics.py | 31 +++++++++++++++
 3 files changed, 129 insertions(+)

diff --git a/arrayfire/__init__.py b/arrayfire/__init__.py
index e44a77f..fff3536 100644
--- a/arrayfire/__init__.py
+++ b/arrayfire/__init__.py
@@ -5,3 +5,4 @@ from .algorithm import *
 from .device import *
 from .blas import *
 from .arith import *
+from .statistics import *
diff --git a/arrayfire/statistics.py b/arrayfire/statistics.py
new file mode 100644
index 0000000..78d547e
--- /dev/null
+++ b/arrayfire/statistics.py
@@ -0,0 +1,97 @@
+from .library import *
+from .array import *
+
+def mean(a, weights=None, dim=None):
+    if dim is not None:
+        out = array()
+
+        if weights is None:
+            safe_call(clib.af_mean(pointer(out.arr), a.arr, c_int(dim)))
+        else:
+            safe_call(clib.af_mean_weighted(pointer(out.arr), a.arr, weights.arr, c_int(dim)))
+
+        return out
+    else:
+        real = c_double(0)
+        imag = c_double(0)
+
+        if weights is None:
+            safe_call(clib.af_mean_all(pointer(real), pointer(imag), a.arr))
+        else:
+            safe_call(clib.af_mean_all_weighted(pointer(real), pointer(imag), a.arr, weights.arr))
+
+        real = real.value
+        imag = imag.value
+
+        return real if imag == 0 else real + imag * 1j
+
+def var(a, isbiased=False, weights=None, dim=None):
+    if dim is not None:
+        out = array()
+
+        if weights is None:
+            safe_call(clib.af_var(pointer(out.arr), a.arr, isbiased, c_int(dim)))
+        else:
+            safe_call(clib.af_var_weighted(pointer(out.arr), a.arr, weights.arr, c_int(dim)))
+
+        return out
+    else:
+        real = c_double(0)
+        imag = c_double(0)
+
+        if weights is None:
+            safe_call(clib.af_var_all(pointer(real), pointer(imag), a.arr, isbiased))
+        else:
+            safe_call(clib.af_var_all_weighted(pointer(real), pointer(imag), a.arr, weights.arr))
+
+        real = real.value
+        imag = imag.value
+
+        return real if imag == 0 else real + imag * 1j
+
+def stdev(a, dim=None):
+    if dim is not None:
+        out = array()
+        safe_call(clib.af_stdev(pointer(out.arr), a.arr, c_int(dim)))
+        return out
+    else:
+        real = c_double(0)
+        imag = c_double(0)
+        safe_call(clib.af_stdev_all(pointer(real), pointer(imag), a.arr))
+        real = real.value
+        imag = imag.value
+        return real if imag == 0 else real + imag * 1j
+
+def cov(a, isbiased=False, dim=None):
+    if dim is not None:
+        out = array()
+        safe_call(clib.af_cov(pointer(out.arr), a.arr, isbiased, c_int(dim)))
+        return out
+    else:
+        real = c_double(0)
+        imag = c_double(0)
+        safe_call(clib.af_cov_all(pointer(real), pointer(imag), a.arr, isbiased))
+        real = real.value
+        imag = imag.value
+        return real if imag == 0 else real + imag * 1j
+
+def median(a, dim=None):
+    if dim is not None:
+        out = array()
+        safe_call(clib.af_median(pointer(out.arr), a.arr, c_int(dim)))
+        return out
+    else:
+        real = c_double(0)
+        imag = c_double(0)
+        safe_call(clib.af_median_all(pointer(real), pointer(imag), a.arr))
+        real = real.value
+        imag = imag.value
+        return real if imag == 0 else real + imag * 1j
+
+def corrcoef(x, y):
+    real = c_double(0)
+    imag = c_double(0)
+    safe_call(clib.af_corrcoef(pointer(real), pointer(imag), x.arr, y.arr))
+    real = real.value
+    imag = imag.value
+    return real if imag == 0 else real + imag * 1j
diff --git a/tests/simple_statistics.py b/tests/simple_statistics.py
new file mode 100644
index 0000000..c2c0301
--- /dev/null
+++ b/tests/simple_statistics.py
@@ -0,0 +1,31 @@
+#!/usr/bin/python
+import arrayfire as af
+
+a = af.randu(5, 5)
+b = af.randu(5, 5)
+w = af.randu(5, 1)
+
+af.print_array(af.mean(a, dim=0))
+af.print_array(af.mean(a, weights=w, dim=0))
+print(af.mean(a))
+print(af.mean(a, weights=w))
+
+af.print_array(af.var(a, dim=0))
+af.print_array(af.var(a, isbiased=True, dim=0))
+af.print_array(af.var(a, weights=w, dim=0))
+print(af.var(a))
+print(af.var(a, isbiased=True))
+print(af.var(a, weights=w))
+
+af.print_array(af.stdev(a, dim=0))
+print(af.stdev(a))
+
+af.print_array(af.var(a, dim=0))
+af.print_array(af.var(a, isbiased=True, dim=0))
+print(af.var(a))
+print(af.var(a, isbiased=True))
+
+af.print_array(af.median(a, dim=0))
+print(af.median(w))
+
+print(af.corrcoef(a, b))

-- 
Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/debian-science/packages/python-arrayfire.git