gpt4 book ai didi

python - 如何从矩阵的每一行中获取最大值和位置

转载 作者:太空宇宙 更新时间:2023-11-04 04:40:19 24 4
gpt4 key购买 nike

<分区>

假设我有一个这样的矩阵 A=np.array([[1,2,3],[5,2,6],[7,1,5]])

然后,我想从每一行中选择最大值和位置。

结果应该是 Value=[3,6,7], Position=[2,2,0]。

在Matlab中,代码[Value,Position]=max(A);可以计算出正确答案

但是我需要把它改成Python代码。

我曾经尝试过这样的代码

Value=np.max(A, axis=1)
Position=np.where(A==np.max(A,axis=1))

Result:
Value=array([3, 6, 7])
Position=(array([], dtype=int32), array([], dtype=int32))

最大值正确,但位置错误。

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