ruby - 在 ruby​​ 中合并或求和 "keys"上的 2 个数组

Question

这是数组版本:Sum 2 hashes attributes with the same key

我有 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]]