ruby - 如何使用 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