Python Numpy确定包含整体最大/最小值的数组索引

问题描述 投票:0回答:1

说我有一个Numpy nd数组,如下所示。

arr = np.array([[2, 3, 9], [1,6,7], [4, 5, 8]])

numpy数组是一个3 x 3矩阵。

我想做的是包含总最大值的查找行索引。

例如,在上面的示例中,整个ndarray的最大值为9。由于它位于第一行,所以我想返回索引0。

最理想的方法是什么?

我知道

np.argmax(arr)

可以返回整个数组,每行或每列的最大值的索引。但是,有没有一种方法可以返回包含总最大值的行的索引?

如果也可以轻松地将其更改为按列或查找最小值,这也很好,等等。>

我不确定如何在不使用循环的情况下进行此操作,而在为索引保留变量的同时找到总最大值,但是我觉得这对于numpy效率不高,因此必须有更好的方法。

任何帮助或想法都会受到赞赏。

谢谢您的阅读。

说我有一个Numpy nd数组,如下所示。 arr = np.array([[2,3,9],[1,6,7],[4,5,8]])numpy数组是一个3 x 3的矩阵。我想做的是查找行索引,其中包含整体最大值...

python arrays numpy max min
1个回答
1
投票

whereamaxamin一起使用。

© www.soinside.com 2019 - 2024. All rights reserved.