[解決済み] NumPyの配列の中で、1つの軸に沿った最大要素のインデックスを取得する方法

質問

2次元のNumPy配列を持っています。私は軸上の最大値を取得する方法を知っています。

>>> a = array([[1,2,3],[4,3,1]])
>>> amax(a,axis=0)
array([4, 3, 3])

最大要素のインデックスを取得するにはどうすればよいですか？私は出力としてたいと思います array([1,1,0]) の代わりに

どのように解決するのですか？

>>> a.argmax(axis=0)

array([1, 1, 0])