Skip to content
Snippets Groups Projects
Commit 0a3aad74 authored by Tamas Gal's avatar Tamas Gal :speech_balloon:
Browse files

Add alternative masking

parent 6b96db32
No related branches found
No related tags found
1 merge request!47Resolve "uproot4 integration"
Pipeline #16258 failed
......@@ -323,6 +323,12 @@ def mask(arr, sequence=None, startend=None, minmax=None, atleast=None):
return builder.snapshot()
def mask_alt(arr, start, end):
nonempty = ak.num(arr, axis=-1) > 0
mask = (arr.mask[nonempty][..., 0] == start) & (arr.mask[nonempty][..., -1])
return ak.fill_none(mask, False)
@nb.njit
def _mask3d(arr, builder, sequence=None, startend=None, minmax=None, atleast=None):
for subarray in arr:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment