How to filter a numpy array with another array's values,a[f]