Julia, for循环迭代器递增到零

Question

我有一个很好的函数，它对由该函数生成的一组随机数求和。

sum(round(Int8, floor(
rand(TruncatedNormal(var1, var2, var3, var4))))
for _ in 1:var5)

大多数情况下，它工作得很好，var5,for循环的迭代器在函数中较早计算，并且可以合法地为零。因此，通过从1增加到0的迭代不起作用，并抛出错误。有干净的"支票"吗?在Julia中，这将简单地为整个函数产生0的结果，或者我是否需要返回并将该函数包含在if语句中以检查var5的值，然后再运行它，如果var5<=0，则产生替代结果(0)?谢谢。J

Answer 1

当您执行sum(rand(Int8) for _ in 1:0)时，您会得到错误:

julia> sum(rand(Int8) for x in 1:0)
ERROR: ArgumentError: reducing over an empty collection is not allowed
Stacktrace:
[1] _empty_reduce_error() at ./reduce.jl:299
[2] mapreduce_empty(::Function, ::Base.BottomRF{typeof(Base.add_sum)}, ::Type{T} where T) at ./reduce.jl:342
[3] reduce_empty(::Base.MappingRF{var"#9#10",Base.BottomRF{typeof(Base.add_sum)}}, ::Type{Int64}) at ./reduce.jl:329
[4] reduce_empty_iter at ./reduce.jl:355 [inlined]
[5] reduce_empty_iter at ./reduce.jl:354 [inlined]
[6] foldl_impl at ./reduce.jl:49 [inlined]
[7] mapfoldl_impl at ./reduce.jl:44 [inlined]
[8] #mapfoldl#204 at ./reduce.jl:160 [inlined]
[9] mapfoldl at ./reduce.jl:160 [inlined]
[10] #mapreduce#208 at ./reduce.jl:287 [inlined]
[11] mapreduce at ./reduce.jl:287 [inlined]
[12] sum at ./reduce.jl:494 [inlined]
[13] sum(::Base.Generator{UnitRange{Int64},var"#9#10"}) at ./reduce.jl:511
[14] top-level scope at REPL[6]:1

一个非常简单的解决方法是首先将随机数收集到一个数组中-或者简单地使用推导式而不是生成器:

julia> sum([rand(Int8) for x in 1:0])
0

您也可以使用上面链接的三元表达式来简单地完全避免这个问题:

julia> var5 = 0
0
julia> var5 < 1 ? 0 : sum(rand(Int8) for x in 1:var5)
0
julia> var5 = 2
2
julia> var5 < 1 ? 0 : sum(rand(Int8) for x in 1:var5)
49

Answer 2

对于需要捕获对空集合求和的情况，我更倾向于使用具有init值的显式reduce:

julia> reduce(+, (rand(Int8) for x in 1:0), init=Int8(0))
0