如何沿一个轴获取numpy数组中最大元素的索引
我有一个2维NumPy数组。我知道如何获得轴上的最大值:
>>> a = array([[1,2,3],[4,3,1]])
>>> amax(a,axis=0)
array([4, 3, 3])
如何获得最大元素的索引?所以我想输出array([1,1,0])
没有找到相关结果
已邀请:
4 个回复
完趣镐
栖很钾是狠
嫩昧竞莫
只返回每行的第一次出现。 http://docs.scipy.org/doc/numpy/reference/generated/numpy.argmax.html 如果您需要为形状阵列执行此操作,这比
更好:
您还可以更改条件:
以上以您要求的形式提供结果。或者,您可以通过以下方式转换为x,y坐标列表:
窃誓额