在另一个数组中查找哈希并在 Ruby 中减去

Question

我有两个哈希数组，如下所示：

a = [ { car_id: 1, motor_id: 1, quantity: 5 },
      { car_id: 1, motor_id: 2, quantity: 6 },
      { car_id: 5, motor_id: 3, quantity: 3 } ]

b = [ { car_id: 1, motor_id: 1, quantity: 2 },
      { car_id: 1, motor_id: 2, quantity: 3 } ]

对于具有相同 car_id 和 motor_id 的那些哈希，我想从 a 中的哈希中减去 b 中每个哈希的数量。所以，我的预期结果是：

c = [ {car_id: 1, motor_id: 1, quantity: 3},
      {car_id: 1, motor_id: 2, quantity: 3},
      {car_id: 5, motor_id: 3, quantity: 3 } ]

在 Ruby 中执行此操作的好方法是什么？我的想法是遍历a，并为每个元素查找b 中是否有任何具有相同car_id 和motor_id 的元素，如果是，则减去并继续。

Answer 1

我建议您首先为b 创建一个哈希bqty，对于b 的每个元素（哈希）g，将[g[:car_id], g[:motor_id]] 映射到g[:quantity]：

bqty = b.each_with_object({}) {|g,h| h[[g[:car_id], g[:motor_id]]] = g[:quantity]}
  #=> {[1, 1]=>2, [1, 2]=>3}

接下来，将a 的每个元素（哈希）g 映射到所需的哈希。这是通过将g 合并到一个空的哈希h（或g.dup）中来完成的，然后，如果存在bqty 的元素和键key = [h[:car_id], h[:motor_id]]，则从h[:quantity] 中减去bqty[key]。请注意，这会使 a 和 b 保持不变。

a.map do |g|
  {}.merge(g).tap do |h|
    key = [h[:car_id], h[:motor_id]]
    h[:quantity] -= bqty[key] if bqty.key?(key)
  end
end
  #=> [{:car_id=>1, :motor_id=>1, :quantity=>3},
  #    {:car_id=>1, :motor_id=>2, :quantity=>3},
  #    {:car_id=>5, :motor_id=>3, :quantity=>3}]

antepenultimate¹ 行的替代方案是：

    h[:quantity] -= bqty[key].to_i

自nil.to_i #=> 0.

^{1.怎么能错过使用这样一个词的机会？}

Answer 2

这基本上是您的建议，除了对 car_id/motor_id 的显式检查，它使用散列代替。

ids = ->(h){h.values_at(:car_id, :motor_id)} # A function for extracting the key values
lookup = Hash[*a.flat_map{|h| [ids[h], h]}] # A map from ids -> hash w/ quantity
b.each{|h| lookup[ids[h]][:quantity] -= h[:quantity]} # Subtract each b from an a (will error if the a doesn't exist)
a == c # == true

Answer 3

也许，您会发现以下更直观：

    matched_result = []
    while current_a = a.shift
      if current_b = b.shift
        if current_a[:car_id] == current_b[:car_id]  && current_a[:motor_id] == current_b[:motor_id]
          matched_result << {car_id: current_a[:car_id], motor_id: current_a[:motor_id], quantity: current_a[:quantity] - current_b[:quantity]}
        end
      else
        matched_result << current_a
      end
    end
    p matched_result