NumPy内置函数,用于查找矩阵中最大的向量

计算 sum 每行,然后用以下公式获取最大值的位置 argmax 并将其用于索引:

out = arr[arr.sum(axis=1).argmax()]

输出: array([5, 4])

中间体:

arr.sum(axis=1)
# array([3, 6, 6, 9])

arr.sum(axis=1).argmax()
# 3

变体:所有最大值

如果可以有多个最大值,则可以使用:

out = arr[(s:=arr.sum(axis=1)) == s.max()]

或者对于python<3.8:

s = arr.sum(axis=1)
out = arr[s == s.max()]

输出:

array([[5, 4]])

注意2D输出,如果几行具有相同的最大和,则会得到多行。