合并哈希数组：值应该是合并值的平均值

Question

问题：将具有相同值的哈希数组与指定键的值合并，并找到其他键的平均值。

我的解决方案似乎很难看

数据：

require 'pp'

arr = [{:red=>346.0,
  :unu=>10.0,
  :used=>20147.0,
  :acc_id=>550,
  :percent=>0.01},
 {:red=>0.0,
  :unu=>1.0,
  :used=>66.0,
  :acc_id=>550,
  :percent=>0.06},
 {:red=>120.0,
  :unu=>11.0,
  :used=>166.0,
  :acc_id=>550,
  :percent=>10.06},
 {:red=>1306.0,
  :unu=>1.0,
  :used=>13259.0,
  :acc_id=>9999,
  :percent=>0.0}]

在当前示例中，我们应该将 3 个哈希与 (:acc_id = 550) 合并，结果数组应包含两个哈希（合并哈希与 :acc_id = 550 和未触及哈希与 :acc_id = 9999）

算法：

data = []
arr.group_by{|h| h[:acc_id] }.map {|_, arr_of_hashes|
  sz = arr_of_hashes.size
  if sz > 1
    arr_of_hashes = arr_of_hashes.inject{|memo, el|
      memo.merge(el) {|k, old_v, new_v| old_v + new_v}
    }

    arr_of_hashes.map {|k, v| arr_of_hashes[k] = v / sz}
  end
  data << arr_of_hashes if arr_of_hashes.is_a? Hash
  data << arr_of_hashes[0] if arr_of_hashes.is_a? Array
}

pp data

预期结果： 合并哈希数组

[{:red=>155.33333333333334,
  :unu=>7.333333333333333,
  :used=>6793.0,
  :acc_id=>550,
  :percent=>3.376666666666667},
 {:red=>1306.0,
  :unu=>1.0,
  :used=>13259.0,
  :acc_id=>9999,
  :percent=>0.0}]

... ... ...

Answer 1

我发现你在acc_id 上执行/ 和+ 的一个错误已经重构了代码 让我们试试这个，但我想我们仍然会改进它。

data = []
arr.group_by{|h| h[:acc_id] }.map {|_, arr_of_hashes|
  sz = arr_of_hashes.size
  result = Hash.new(0)
  arr_of_hashes.map{ |hash| hash.map{ |k,v| result[k] += v/sz unless k == :acc_id } }
  result[:acc_id] = arr_of_hashes.first[:acc_id]
  data << result
}

Answer 2

我建议您按如下方式计算平均值。我假设，与示例一样，所有哈希都具有相同的键（尽管它们不一定以相同的顺序出现）。

代码

def doit(arr, key)
  keys = arr.first.keys
  arr.group_by { |g| g[key] }.
      map do |_,a| 
        averages = a.map { |h| h.values_at(*keys) }.
                     transpose.
                     map { |v| v.sum.fdiv(v.size) }
        keys.zip(averages).to_h
      end
end 

示例

请注意，我的哈希数组与问题示例中给出的哈希数组有些不同。具体来说，有三组（而不是两组）哈希，其中键 :acc_id 的值具有共同值。

arr = [{ red: 346.0,  unu: 10.0, used: 20147.0, acc_id: 550,  percent: 0.01 },
       { red: 0.0,    unu: 1.0,  used: 66.0,    acc_id: 550,  percent: 0.06 },
       { red: 120.0,  unu: 11.0, used: 166.0,   acc_id: 10,   percent: 10.06 },
       { red: 100.0,  unu: 19.0, used: 170.0,   acc_id: 10,   percent: 11.56 },
       { red: 1306.0, unu: 1.0,  used: 13259.0, acc_id: 9999, percent: 0.0 }]

doit(arr, :acc_id)
  #=> [{:red=>173.0, :unu=>5.5, :used=>10106.5, :acc_id=>550.0, :percent=>0.035},
  #    {:red=>110.0, :unu=>15.0, :used=>168.0, :acc_id=>10.0, :percent=>10.81},
  #    {:red=>1306.0, :unu=>1.0, :used=>13259.0, :acc_id=>9999.0, :percent=>0.0}]

说明

参见Enumerable#group_by 和Array#sum（后者在v2.4 中首次亮相）。

步骤如下。

key = :acc_id
keys = arr.first.keys
  #=> [:red, :unu, :used, :acc_id, :percent]
b = arr.group_by { |g| g[key] }
  #=> { 550=>[{:red=>346.0, :unu=>10.0, :used=>20147.0, :acc_id=>550, :percent=>0.01},
  #           {:red=>0.0, :unu=>1.0, :used=>66.0, :acc_id=>550, :percent=>0.06}],
  #      10=>[{:red=>120.0, :unu=>11.0, :used=>166.0, :acc_id=>10, :percent=>10.06},
  #           {:red=>100.0, :unu=>19.0, :used=>170.0, :acc_id=>10, :percent=>11.56}],
  #    9999=>[{:red=>1306.0, :unu=>1.0, :used=>13259.0, :acc_id=>9999, :percent=>0.0}]}
b.map do |_,a| 
  averages = a.map { |h| h.values_at(*keys) }.
               transpose.
               map { |v| v.sum.fdiv(v.size) }
  keys.zip(averages).to_h
end 
  #=> <the array of hashes shown above>