De uma codificação quente para python inteiro

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