От одного горячего кодирования до Integer Python

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