（Julia）快速对数组求列和

Question

我有一个浮点数的 MxN 数组 A，我想进行以下操作：对于 A 的每一列，计算小于某个阈值（例如 0.5）的元素数量。

Julia执行此操作的最快方法是使用零初始化结果向量，然后按列遍历数组A并在需要时递增结果向量。例如，使用for循环执行此操作很容易

function count(A)
    (cols, rows) = size(A)
    result = fill(0, cols)
    for j in 1:rows
        for i in 1:cols
            result[i] += A[i,j] <= 0.5
        end
    end
end

这将按照与内存中的布局相同的顺序遍历 A，并且不会分配不必要的额外空间。但是，我不确定如何使用例如广播运营商来实现这一点。

sum(A .<= 0.5, dims=1)

但是这只是为了执行操作而分配新内存，并且比Count-函数慢2-3倍（我测试了高达8000x8000的数组）。有没有可能在只使用几行代码的情况下实现Count-函数的性能（如第二种方法）？

Answer 1

我写你病情的一种自然方式是：

count.(<=(0.5), eachcol(A))

或

vec(sum(<=(0.5), A, dims=1))

（后者比前者稍慢，但执行的分配要少得多，因此性能平衡可能取决于矩阵的维度）

一般来说，两者都应该相当快速且易于阅读。

作为旁注 - 您的代码在行中而不是列中执行计数。这是更正它以在列中执行计数的方法：

function count2(A)
    (rows, cols) = size(A)
    result = fill(0, cols)
    for i in 1:cols
        for j in 1:rows
            result[i] += A[j,i] <= 0.5
        end
    end
    result
end

最后注意和（A.