https://tutorial.eyehunts.com/python/numpy-filter-array-by-condition-example-code/