mirror of https://github.com/pyodide/pyodide.git
18 lines
572 B
Python
18 lines
572 B
Python
# setup: n=100 ; import numpy; db = numpy.random.randint(2, size=(n, 4), dtype='bool') # noqa
|
|
# run: check_mask(db)
|
|
# from:
|
|
# http://stackoverflow.com/questions/34500913/numba-slower-for-numpy-bitwise-and-on-boolean-arrays
|
|
|
|
# pythran export check_mask(bool[][])
|
|
import numpy as np
|
|
|
|
|
|
def check_mask(db, mask=[1, 0, 1]):
|
|
out = np.zeros(db.shape[0], dtype=bool)
|
|
for idx, line in enumerate(db):
|
|
target, vector = line[0], line[1:]
|
|
if (mask == np.bitwise_and(mask, vector)).all():
|
|
if target == 1:
|
|
out[idx] = 1
|
|
return out
|