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

Question

我的数据如下-

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

我想找到包含最大值的行。在本例中，它是1（索引为0，因此为1）

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

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

我想知道有没有别的办法

我使用的是python2.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

Answer 2

使用np.argmax(x)获得{}后，可以执行以下操作：

>>> np.unravel_index(np.argmax(x), x.shape)
(1, 3)

这会将平面索引转换回二维索引，您可以很容易地看到它是哪一行。在