如何按年/月和总值对ruby literal hash进行分组

Question

我有以下哈希，我想按年/月分组并对值求和：

range = {
"2020-11-06 00:00:00 +0000" => 234100.14176395803,
"2020-11-07 00:00:00 +0000" => 57731.63072153537,
"2020-11-08 00:00:00 +0000" => 68903.8902730576,
"2020-11-09 00:00:00 +0000" => 180971.98008691627,
........
"2021-02-01 00:00:00 +0000" => 169004.96299567595,
"2021-02-02 00:00:00 +0000" => 183363.9687217272,
"2021-02-03 00:00:00 +0000" => 200400.2505103338
}

我可以进行分组：hash = Hash[range.collect { |k,v| [k.to_date.strftime("%Y-%m"), v]}]，但我无法获得值的总和、值的结果，并且使用哈希收集只是最后一天。

{
"2020-11" => 155346.83103528828,
"2020-12" => 175424.41029800082,
"2021-01" => 55366.44438577791,
"2021-02" => 200400.2505103338
}

使用reduce方法hash.values.reduce(:+)，我可以得到总值，但我需要逐年按月计算。

Answer 1

首先我们需要将其分组为几个月。使用.group_by可以最容易地做到这一点。

hash = Hash[range.group_by { |k,_v| k.to_date.strftime("%Y-%m")}]

使用您已经提供的相同代码，只使用group_by而不是collect。

{
"2020-11"=>[
["2020-11-06 00:00:00 +0000", 234100.14176395803],
["2020-11-07 00:00:00 +0000", 57731.63072153537],
["2020-11-08 00:00:00 +0000", 68903.8902730576],
["2020-11-09 00:00:00 +0000", 180971.98008691627]
],
"2021-02"=>[
["2021-02-01 00:00:00 +0000", 169004.96299567595],
["2021-02-02 00:00:00 +0000", 183363.9687217272],
["2021-02-03 00:00:00 +0000", 200400.2505103338]
]
}

现在，我们需要将sum(相当于您尝试的reduce(:+)，但有一些好处(它们放在一起。我们需要修改哈希的每个值，并对值中每个元素的最后一个元素求和。

hash.transform_values {|a| a.sum(&:last) }

返回

{
"2020-11" => 541707.6428454673,
"2021-02" => 552769.1822277369
}

Answer 2

您可以通过each_with_object:构建新的哈希

range.each_with_object(Hash.new(0)) { |(k, v), h| h[k[0, 7]] += v }
#=> {"2020-11"=>541707.6428454672, ..., "2021-02"=>552769.1822277369}

代替k[0, 7]，您也可以使用k.to_date.strftime('%Y-%m')