Od jednego gorącego kodowania do Pythona całkowitego

np.argmax(a, axis=1): [1 0 3]
Nutty Nightingale