For example:
arr = np.array([[0,1],[0,2],[0,3],[1,4],[1,5],[2,6],[3,7]]) arr2 = arr[arr[:,0]==1]
In this example its very fast as is, but if