de una codificación caliente a entero python

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