Merge pull request #145 from scikit-hep/issue-144

Work-around Numpy's ufunc.reduceat
scikit-hep · Jun 17, 2019 · dcab91f · dcab91f
2 parents 1c3cf87 + 827d9fe
commit dcab91f
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/awkward/array/jagged.py b/awkward/array/jagged.py
@@ -1444,6 +1444,11 @@ def _reduce(self, ufunc, identity, dtype, regularaxis):
                     content = ufunc.reduce(content, axis=axis)
 
             out = ufunc.reduceat(content, nonterminal)[:len(out)]
+            if len(out) < len(thyself):
+                tmp = self.numpy.empty((len(thyself),) + out.shape[1:], dtype=out.dtype)
+                tmp[:len(out)] = out
+                out = tmp
+
             out[thyself.starts == thyself.stops] = identity
 
         if regularaxis is None:

diff --git a/awkward/version.py b/awkward/version.py
@@ -4,7 +4,7 @@
 
 import re
 
-__version__ = "0.11.0"
+__version__ = "0.11.1"
 version = __version__
 version_info = tuple(re.split(r"[-\.]", __version__))
 

diff --git a/tests/test_issues.py b/tests/test_issues.py
@@ -17,3 +17,9 @@ def test_issue49(self):
         a = JaggedArray([2], [5], [1, 2, 3, 4, 5])
         m = JaggedArray([0], [3], [False, False, True])
         assert a[m].tolist() == [[5]]
+
+    def test_issue144(self):
+        x = fromiter([[], [0.5], [0.6], []])
+        isloose = fromiter([[], [False], [True], []])
+        assert x[isloose].tolist() == [[], [], [0.6], []]
+        assert x.sum().tolist() == [0.0, 0.5, 0.6, 0.0]