複数の配列をマージする方法

Question

初歩的な質問となってしまいますが、よろしくお願い致します。
以下のようなa、bをcの配列にしたいです。

a = [{index: 1, kingaku+: 100},
     {index: 2, kingaku+: 200},
     {index: 3, kingaku+: 300}] 

b = [{index: 1, kingaku-:50},
     {index: 3, kingaku-:50}]

c = [{index: 1, kingaku+: 100, kingaku-: 50, kingakudif: 50},
     {index: 2, kingaku+: 200, kingaku-: 0, kingakudif: 200},
     {index: 3, kingaku+: 300, kingaku-: 50, kingakudif: 250}] 

思いつく方法としては、aをぐるぐるまわして泥臭く計算するようなロジックですが、railsのメソッド等を使って綺麗にかけないかと思って質問いたしました。
ご教示いただきたいです。

RubyonRails5

Answer 1

railsであればActiveSupportのEnumarable#index_by(&)が使えるので、こんな風にできますかねぇ。
分かりやすさと拡張性優先なので、遅いかも。

ax = a.index_by {|ia| ia[:index] }
bx = b.index_by {|ib| ib[:index] }
items = ax.merge(bx) do |_, ia, ib|
  i = ia.merge(ib)
  i[:'kingaku+'] ||= 0
  i[:'kingaku-'] ||= 0
  i[:kingakudif] = i[:'kingaku+'] - i[:'kingaku-']
  i
end.values

余談ですが、こういう処理はどちらかというとRDBの得意分野なので
RailsならDBレイヤーで処理したほうが良いような気はします。

Comments

[hiepita1]

回答ありがとうございます。速度等も検証してみます！参考にさせていただきます。

Answer 2

a_plus =  [
  {index: 1, 'kingaku+' => 100},
  {index: 2, 'kingaku+' => 200},
  {index: 3, 'kingaku+' => 300}
].map{|item| [ item[:index], item['kingaku+']].to_h

b_minus = [
  {index: 1, 'kingaku-' => 50},
  {index: 3, 'kingaku-' => 50}
].map{|item| [ item[:index], item['kingaku-']].to_h


[a_plus.keys + b_minus.keys].uniq.map do |index|
  plus = a_plus[index].presence || 0　#存在していたらその値、ないなら０になる定番構文
  minus = b_minus[index].presence || 0
  {index: index, kingaku_plus: plus, kingaku_minus: minus, kingaku_dif: plus- minus}
end

こんな感じでいいのでは。
ポイントは、map.to_h による　Hash化
Hashは、アクセススピードが速いので気軽に使える。自分で調べてね。

あと、hashのkeyに＋－の記号を混ぜるのはやめろー。面倒だし気持ち悪い。

Comments

[hiepita1]

回答ありがとうございます。最小限のコード量で実現出来ているなと感じました。
参考にさせていただきます。

Answer 3

泥臭いですが、動作するものを書いてみました。
calc.rb

def sub_total(rec, attr)
  adds = rec["#{attr}+"] == nil ? 0 : rec["#{attr}+"].sum
  subs = rec["#{attr}-"] == nil ? 0 : rec["#{attr}-"].sum
  {"#{attr}+" => adds, "#{attr}-" => subs, "#{attr}dif" => adds - subs}
end

def total(a, b, attr)
  ab = (a + b).each_with_object({}) do |rec, mem|
    idx = rec.delete(:index)
    rec.each do |kind, val|
      mem[idx] = {} if mem[idx] == nil
      mem[idx][kind] = [] if mem[idx][kind] == nil
      mem[idx][kind] += [val]
    end
  end
  # p "ab"
  # p ab

  ab.each_with_object([]) do |(idx, rec), mem|
    mem << {index: idx}.merge(sub_total(rec, attr))
  end
end

a = [
  {index: 1, 'kingaku+' => 100},
  {index: 2, 'kingaku+' => 200},
  {index: 3, 'kingaku+' => 300}
]
b = [
  {index: 1, 'kingaku-' => 50},
  {index: 3, 'kingaku-' => 50}
]
p total(a, b, 'kingaku')

実行結果

Comments

[hiepita1]

回答ありがとうございます。参考にさせていただきます。