Learn how to index a numpy array with a boolean array for python programming