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)