How can I index this numpy array? [duplicate]

Answer 1

Use arange for the other dimension:

c = a[np.arange(5), b]