arrays - 检查Julia数组中的所有元素是否相等

Question

我可以想到的测试数组arr中所有元素是否相等的最短方法是all(arr[1] .== arr)。虽然这肯定很短，但似乎有点不雅。有内置的功能吗？

我怀疑 的含义，但是这行不通，因为 ==(arr...)运算符只能接受两个参数。我不确定Julia如何解析 ==之类的表达式，但是有什么方法可以使它适应具有任意数量元素的数组？

Answer 1

很棒的问题@tparker和很棒的答案@ColinTBowers。在尝试同时考虑它们的同时，我想到了一种直接的老式朱利安for循环方式。在相同元素的长向量的重要输入上，结果更快，因此，我添加了此注释。另外，函数名称allequal似乎很合适。所以这是变体：

allequal_1(x) = all(y->y==x[1],x)

# allequal_2(x) used to be erroneously defined as foldl(==,x)   

@inline function allequal_3(x)
    length(x) < 2 && return true
    e1 = x[1]
    i = 2
    @inbounds for i=2:length(x)
        x[i] == e1 || return false
    end
    return true
end

和基准：

julia> using BenchmarkTools

julia> v = fill(1,10_000_000);  # long vector of 1s

julia> allequal_1(v)
true

julia> allequal_3(v)
true

julia> @btime allequal_1($v);
  9.573 ms (1 allocation: 16 bytes)

julia> @btime allequal_3($v);
  6.853 ms (0 allocations: 0 bytes)

更新：另一个重要的基准案例是发生短路的机会。因此（根据要求）：

julia> v[100] = 2
2

julia> allequal_1(v),allequal_2(v),allequal_3(v)
(false, false, false)

julia> @btime allequal_1($v);
  108.946 ns (1 allocation: 16 bytes)

julia> @btime allequal_3($v);
  68.221 ns (0 allocations: 0 bytes)

在所有条件都相同的情况下， for版本在Base中应为 allequal。