Selecting specific column in each row from array,"a[np.arange(3), (0, 1, 0)]"