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