gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-11-30 23:18:59 61 4
gpt4 key购买 nike

我的数据如下 -

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

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

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

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

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

我正在使用Python 2.7。谢谢。

最佳答案

试试这个:

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

关于python - 在 numpy 数组中查找最大值行的最安全方法,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/26373837/

61 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com