Ruby 将哈希值转换为运行总和

Question

我有一个哈希

{:a => 2, :b => 8, :c => 10, :d => 40 }

我想修改散列值，使这些值代表运行总计。

{:a => 2, :b => 10, :c => 20, :d => 60 }

我已经走了几条路，但还没有找到正确的答案。我尝试过的一些事情。

data.inject({}) { |result, (k,v)| result[k] = v+ result.values.compact.sum; result }

=> {:a=>2, :b=>10, :c=>22, :d=>74}

我显然不太了解结果是如何在注入中累积的。由于 k 的第一次迭代为零，我正在使用紧凑型。

我可以像这样将值作为数组获取

data.inject([]) { |result, element| result << result.last.to_i + element.last }

=> [2, 10, 20, 60]

我不知道如何修改它以与哈希类似地工作。

我可以这样做，但这似乎不是正确/最有效的方法。

totals = data.inject([]) { |result, element| result << result.last.to_i + element.last }

=> [2, 10, 20, 60]

data.map.with_index { |(k,v),i| [k,totals[i]] }.to_h

=> {:a=>2, :b=>10, :c=>20, :d=>60}

感谢我在这里得到的任何帮助。

Answer 1

鉴于您知道键的顺序，您可以简单地编写

h= { :a => 2, :b => 8, :c => 10, :d => 40 }

cumulative = 0
h.transform_values { |v| cumulative += v }
  #=> {:a=>2, :b=>10, :c=>20, :d=>60}

见Hash#transform_values。

Answer 2

您的“inject 和 map.with_index”很接近，但您可以将它们组合起来：

data.inject({}) { |h, (k, v)| h.merge(k => v + h.values.last.to_i) }

或复制较少：

data.inject({}) { |h, (k, v)| h.merge!(k => v + h.values.last.to_i) }

或许：

data.each_with_object({}) { |(k, v), h| h[k] = v + h.values.last.to_i }

h 开始时为空，因此 h.values 可以为空，因此 h.values.last.nil? 会发生，因此 #to_i 调用“隐藏”nil 测试并简化块。

Answer 3

哈希和排序

不保证哈希是按排序顺序排列的，因此您可以保持运行总计的前提（原则上）是无效的，除非您在将值传递给 #inject 或 #reduce 之前对键进行排序。但是，MRI Ruby 通常会按插入顺序保留哈希，但请自行承担风险。

解决方案

抛开 Hash 的排序顺序，您可以使用 #sort 或 #sort_by 来解决，如果您愿意，还可以进行一些重构，我将在 Ruby 3.0.2 中像这样修改原始 Hash：

hash = {:a => 2, :b => 8, :c => 10, :d => 40 }

# Note that positional argument `_1` is the
# accumulator, and `_2` is the value.
hash = hash.keys.zip(
  hash.values.inject([]) { _1 << _2 + _1&.last.to_i }
).to_h;
hash
#=> {:a=>2, :b=>10, :c=>20, :d=>60}

如果您不想分配回同一个哈希，或者想将结果分配到其他地方，您也可以这样做。但是，据我了解，问题是关于就地修改您的 Hash，将表达式的结果分配回 hash 似乎是最简单的事情。