make bitstring func

rasbt · rasbt · commit 01fbbeb69c19 · 2014-09-23T13:31:37.000-04:00
diff --git a/howtos_as_py_files/make_bitstring.py b/howtos_as_py_files/make_bitstring.py
@@ -0,0 +1,14 @@
+# Generating a bitstring from a Python list or numpy array
+# where all postive values -> 1
+# all negative values -> 0
+
+def make_bitstring(ary)
+    return np.where(ary > 0, 1, 0)
+
+
+### Example:
+
+ary1 = np.array([1, 2, 0.3, -1, -2])
+make_bitstring(ary1)
+
+# returns array([1, 1, 1, 0, 0])
