gpt4 book ai didi

python - Numpy:如何使用 argmax 结果来获得实际最大值?

转载 作者:太空狗 更新时间:2023-10-30 00:58:56 24 4
gpt4 key购买 nike

<分区>

假设我有一个 3D 数组:

>>> a
array([[[7, 0],
[3, 6]],

[[2, 4],
[5, 1]]])

我可以使用 axis=1 获取它的 argmax

>>> m = np.argmax(a, axis=1)
>>> m
array([[0, 1],
[1, 0]])

如何使用 m 作为 a 的索引,以便结果等同于简单地使用 max

>>> a.max(axis=1)
array([[7, 6],
[5, 4]])

(这在m应用于其他相同形状的数组时很有用)

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