如何使用 Ruby 的 Inject 功能计算平均值

Question

我认为 Ruby 的 inject 方法非常适合计算总和。但是如果我必须计算一个平均值并使用inject，我总是会得到这样的结果：

total_ranked_query_results = @results.count { |result| result.class == RankedQueryResult }
@results.inject(0.0) { |sum, result| result.class == RankedResult ? sum + result.average_precision : sum } / total_ranked_query_results

所以我基本上必须迭代两次才能找出有多少RankedResults 才能计算平均值。

有没有更好的（rubyesque）方法来避免这种情况？

Answer 1

在同一迭代中进行计数和求和：

tmp = @results.inject([0.0, 0]) { |sum, result|
    if RankedResult === result
        [sum.first + result.average_precision, sum.last+1]
    else
        sum
    end
}
tmp.first / tmp.last

或者先过滤集合：

tmp = @results.filter { |result| RankedResult === result }
tmp.inject(0.0) { |sum, result| sum+result.avergage_precision } / tmp.length