如何在NumPy数组中沿着一个轴获取最大元素的索引

10 浏览
0 Comments

如何在NumPy数组中沿着一个轴获取最大元素的索引

我有一个二维的NumPy数组。我知道如何在轴上获取最大值：

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

我如何获取最大元素的索引？我希望输出为array([1,1,0])。

0