Julia Sort.partialsortperm用法及代碼示例

partialsortperm(v, k; by=<transform>, lt=<comparison>, rev=false)

返回向量 v 的部分排列 I ，以便 v[I] 在索引 k 處返回 v 的完全排序版本的值。如果k 是一個範圍，則返回一個索引向量；如果 k 是整數，則返回單個索引。使用與 sort! 相同的關鍵字指定順序。排列是穩定的，這意味著相等元素的索引以升序出現。

請注意，此函數等效於調用 sortperm(...)[k] ，但比調用更有效。

例子

julia> v = [3, 1, 2, 1];

julia> v[partialsortperm(v, 1)]
1

julia> p = partialsortperm(v, 1:3)
3-element view(::Vector{Int64}, 1:3) with eltype Int64:
 2
 4
 3

julia> v[p]
3-element Vector{Int64}:
 1
 1
 2