在ruby中的"键"上合并或求和2个数组

Question

这是数组版本:Sum 2哈希属性具有相同的键

我有2个数组,例如:

a = [[1,10],[2,20],[3,30]]
b = [[1,50],[3,70]]

如何在第一个值(如果存在)上对每个值求和:

c = [[1,60],[2,20],[3,100]]

Answer 1

你可以这样做:

(a + b).group_by(&:first).map { |k, v| [k, v.map(&:last).inject(:+)] }

首先,你把数组连同+因为你不关心a和b,你只关心自己的元素.然后group_by通过第一个元素对组合数组进行分区,以便可以轻松处理内部数组.然后你只需要拉出内部数组的第二个(或最后一个)元素v.map(&:last)并将它们相加inject(:+).

例如:

>> a = [[1,10],[2,20],[3,30]]
>> b = [[1,50],[3,70]]
>> (a + b).group_by(&:first).map { |k,v| [k, v.map(&:last).inject(:+)] }
=> [[1, 60], [2, 20], [3, 100]]

Answer 2

您也可以使用哈希方式:

Hash[a].merge(Hash[b]){|k,a,b|a+b}.to_a