Julia argmax用法及代码示例

argmax(r::AbstractRange)

范围可以有多个最大元素。在这种情况下，argmax 将返回一个最大索引，但不一定是第一个。

argmax(f, domain)

在f 的域中返回一个值x，其中f(x) 被最大化。如果f(x) 有多个最大值，则将找到第一个最大值。

domain 必须是非空的可迭代对象。

将值与 isless 进行比较。

例子

julia> argmax(abs, -10:5)
-10

julia> argmax(cos, 0:π/2:2π)
0.0

argmax(itr)

返回集合中最大元素的索引或键。如果有多个最大元素，则返回第一个。

集合不能为空。

将值与 isless 进行比较。

另请参阅： argmin 、 findmax 。

例子

julia> argmax([8, 0.1, -9, pi])
1

julia> argmax([1, 7, 7, 6])
2

julia> argmax([1, 7, 7, NaN])
4

argmax(A; dims) -> indices

对于数组输入，返回给定维度上最大元素的索引。 NaN 被视为大于除 missing 之外的所有其他值。

例子

julia> A = [1.0 2; 3 4]
2×2 Matrix{Float64}:
 1.0  2.0
 3.0  4.0

julia> argmax(A, dims=1)
1×2 Matrix{CartesianIndex{2}}:
 CartesianIndex(2, 1)  CartesianIndex(2, 2)

julia> argmax(A, dims=2)
2×1 Matrix{CartesianIndex{2}}:
 CartesianIndex(1, 2)
 CartesianIndex(2, 2)