julia - 如何将数组中的小值归零？

Question

有没有一种通用的方法将数组中的小值归零？

“小”是指绝对值小于某个阈值的元素，例如 10.0^-5。

编辑:现在，我使用 eachindex 循环。

function sparsify(a, eps)
    for i in eachindex(a)
        if abs(a[i]) < eps
            a[i] = 0
        end 
    end
end

Answer 1

为什么不直接应用掩码和逐元素小于运算符？

>>> x = rand(Float32, 100)
>>> eps = 0.5
>>> x[abs(x) .< eps] = 0

或作为函数(注意该函数在原地修改向量x):

>>> sparsify!(x, eps) = x[abs(x) .< eps] = 0;

您也可以替换 0与 zero(eltype(x))确保它具有与 x 相同的类型.

x .< eps 创建的临时 bool 掩码将比较 x 的每个元素至eps .然后，每个满足该条件的元素都将设置为 0。