python - 在 numpy 数组中查找最大值行的最安全方法

Question

我的数据如下 -

x= np.array( ((0,0,0,0), (0,0,0,2.8) ))

我想找到包含最大值的行。在此示例中，它是 1(0 索引，因此 1)

np.argmax(x) 返回位置为 7

我可以通过使用 argmax 和 ndim 来迂回查找最大值的行号。

我想知道是否还有其他方法

我正在使用Python 2.7。谢谢。

Answer 1

试试这个:

In [7]: import numpy as np

In [8]: x= np.array( ((0,0,0,0), (0,0,0,2.8) ))

In [9]: np.argmax(x.max(axis=1))
Out[9]: 1