https://dosenit.com/python/numpy-array-filter