哈希的红宝石数组基于键合并数组值



我对ruby中的array的CC_1有一些疑问。

示例:

a = [{236=>1}, {236=>1}, {237=>1}]

我想这样获得结果:

a = [{236=>2}, {237=>1}]

如何根据ruby中的key合并值的array

我发现以这种方式可读:

a = [{236=>1}, {236=>1}, {237=>1}]
merge_with_addition = -> x,y {x.merge(y){|_,old,new|old+new}}
p a.reduce(&merge_with_addition) #=> {236=>2, 237=>1}

有几种方法可以做到这一点。这是使用方法hash#Update(aka merge!)的形式的一种,该形式使用块来确定均存在的两个哈希中存在的键值:

a.each_with_object({}) { |g,h|
  h.update(g["order_detail_id"]=>g) { |_,oh,ng|
    { "order_detail_id" =>g["order_detail_id"],
      "quantity"=>oh["quantity"]+ng["quantity"] } } }.values
  #=> [{"order_detail_id"=>236, "quantity"=>2}, 
  #    {"order_detail_id"=>237, "quantity"=>1}] 

另一种方法使用枚举的#group_by:

a.group_by { |h| h["order_detail_id"] }.map { |id, a|
  { "order_detail_id"=>id, "quantity"=>a.reduce(0) { |t,g|
    t + g["quantity"] } } }
  #=> [{"order_detail_id"=>236, "quantity"=>2}, 
  #    {"order_detail_id"=>237, "quantity"=>1}] 

编辑: Monoy询问了如何合并:

a = [{236=>1}, {236=>1}, {237=>1}]

当您使用单个键值对时,首先将它们转换为阵列通常是最容易的,可以轻松访问键和值:

b = a.map(&:to_a)
  #=> [[[236, 1]], [[236, 1]], [[237, 1]]] 

不幸的是,这有三个级别的数组,我们只需要两个。我们可以写:

b = a.map(&:to_a).flatten(1)
  #=> [[236, 1], [236, 1], [237, 1]] 

,但更简单的方法是:

b = a.flat_map(&:to_a)
  #=> [[236, 1], [236, 1], [237, 1]] 

再次,我们可以将数组合并几种不同的方式。这是使用默认值为零的哈希:

b.each_with_object(Hash.new(0)) { |(k,v),h| h[k] += v }
  #=> {236=>2, 237=>1} 

我们有:

enum = b.each_with_object(Hash.new(0))
  #=> #<Enumerator: [[236, 1], [236, 1], [237, 1]]:each_with_object({})> 

我们可以将此枚举器转换为数组,以查看哪些值each将传递到块中:

enum.to_a
  #=> [[[236, 1], {}], [[236, 1], {}], [[237, 1], {}]]

我们可以使用枚举#隔壁获取枚举器的每个值,并将块变量分配给该值。第一个传递给块是:

(k,v),h = enum.next
  #=> [[236, 1], {}] 
k #=> 236 
v #=> 1 
h #=> {} 

所以在块中我们执行:

h[k] += v

评估:

h[236] += 1

这意味着:

h[236] = h[236] + 1

Hash h目前是空的,因此它没有密钥236。因此,表达式右侧的h[236]返回哈希的默认值零:

h[236] = 0 + 1 

现在:

h #=> {236=>1} 

现在将下一个元素传递到块中:

(k,v),h = enum.next
   #=> [[236, 1], {236=>1}] 
k #=> 236 
v #=> 1 
h #=> {236=>1} 

因此,当我们在块中执行表达式时,现在不使用默认值:

h[k] += v
#=> h[k] = h[k] + v
#=> h[236] = h[236] + 1
#=> h[236] = 1 + 1

现在:

h #=> {236=>2}

现在将枚举者的第三个和最后一个元素传递给块:

(k,v),h = enum.next
   #=> [[237, 1], {236=>2}] 
 k #=> 237 
 v #=> 1 
 h #=> {236=>2}
 h[k] += v
   #=> h[k] = h[k] + v
   #=> h[237] = h[237] + 1
   #=> h[236] = 0 + 1 # the default value of 0 applies
 h #=> {236=>2, 237=>1} 

我们完成了。

array = [{234 => 1}, {235 => 1}, {234 => 1}]
hashes = array.map(&:to_a).flatten
  .each_slice(2).with_object(Hash.new(0)) { |(k, v), h|
  h[k] += v
}.each_with_object([]) { |(k, v), a|
  a << { k => v }
}

最新更新