julia - 相当于 Julia 中的 pandas 'clip'

Question

在 pandas 中，有一个 clip 函数(参见 https://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.clip.html )，它将值限制在用户提供的上下限内。什么是 Julia 等价物？即，我想要:

> clip.([2 3 5 10],3,5)
> [3 3 5 5]

显然，我可以自己编写，或者使用 min 和 max 的组合，但我惊讶地发现没有。 StatsBase 提供 trim 和 winsor 函数，但它们不允许输入固定值，而是允许使用计数或百分位数 (https://juliastats.github.io/StatsBase.jl/stable/robust.html)。

Answer 1

您可能正在寻找clamp:

help?> clamp

  clamp(x, lo, hi)

  Return x if lo <= x <= hi. If x > hi, return hi. If x < lo, return lo. Arguments are promoted to a common type.

这是一个标量 x 的函数，但我们可以使用点符号在向量上广播它:

julia> clamp.([2, 3, 5, 10], 3, 5)
4-element Array{Int64,1}:
 3
 3
 5
 5

如果您不关心原始数组，您也可以使用就地版本 clamp!，它会修改输入:

julia> A = [2, 3, 5, 10];

julia> clamp!(A, 3, 5);

julia> A
4-element Array{Int64,1}:
 3
 3
 5
 5