Boolean array masks in Javascript

Viewed 3522

Coming from Python and Numpy, a typical feature I find myself using frequently are boolean masks.

Here's an example in Python:

>>> mylist = np.array([50, 12, 100, -5, 73])
>>> mylist == 12
array([False, True, False, False, False])  # A new array that is the result of ..
                                           # .. comparing each element to 12
>>> mylist > 0
array([True, True, True, False, True])     # A new array that is the result of ..
                                           # .. comparing each element to 0
>>> mylist[mylist == 12]
array([12])                                # A new array of all values at indexes ..
                                           # .. where the result of `mylist == 12` is True

>>> mask = mylist != 100                   # Save a mask
>>> map(foo, mylist[mask])                 # Apply `foo` where the mask is truthy

In general when np.array is indexed by another array of the same size, a new array is returned containing the elements at those indexes where the mask array's value is truthy.

I am able to do something similar with Array.prototype.map and Array.prototype.filter in Javascript but it's more verbose and my mask is destroyed.

-> mylist = [50, 12, 100, -5, 73]
-> mylist.map(item => item == 12)
<- [false, true, false, false, false]      // mylist == 12

-> mylist.filter(item => item == 12)
<- [12]                                    // mylist[mylist == 12]

-> mask = mylist.map(item => item == 12)
-> mylist.filter(item => mask.unshift())
<- [12]                                    // mylist[mask]

-> mask
<- []                                      // Mask isn't reusable

Is there a better way of applying masks over arrays in javascript or am I stuck making copies of masks and using filter and map each time?

1 Answers
Related