说我有一个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的矩阵。我想做的是查找行索引,其中包含整体最大值...