Numpy Array - Advanced slicing using sum of a one hot encoded column

Question

I\'m trying to slice an array based on a one-hot encoded column, so for an array like this:

import numpy as np
arr = np.array([[0.1,1,0,0],[0.2,1,0,0],[0.3,1,