Rubyでハッシュ値を特定の条件で変換したい

Question

{
  :a => [
    {:id => "1"},
    {:id => "2"},
    {:id => "5"}, # <= id が 2から5へと+3増えている。
    {:id => "6"},
    {:id => "9"}, # <= id が 6から9へと+3増えている。
    {:id => "10"}
  ],
  :b => [
    {:id => "1"},
    {:id => "2"}
  ]
}

Rubyで上のようなhashを、下のように変換したいです。

変換の条件は、配列に含まれるhash({:id=> "1"}など)のid値が一つ前のhashのidと比べて1より多く増加している場合に、自身が所属する配列のkey名(aなど)に_${num}を付けた値をkeyとして分離したいです。

{
  :a => [
    {:id=> "1"},
    {:id=> "2"}
  ],
  :a_1 => [
    {:id=> "5"},
    {:id=> "6"}
  ],
  :a_2 => [
    {:id=> "9"},
    {:id=> "10"}
  ],
  :b => [
    {:id=> "1"},
    {:id=> "2"}
  ]
}

どうぞよろしくお願いいたします。
(疑問などありましたら気兼ねなくご質問お願いしますm)

Answer 1

chunk_while（偽になったところで分割）で。

require "pp"
data = {
  :a => [
    {:id => "1"},
    {:id => "2"},
    {:id => "5"}, # <= id が 2から5へと+3増えている。
    {:id => "6"},
    {:id => "9"}, # <= id が 6から9へと+3増えている。
    {:id => "10"}
  ],
  :b => [
    {:id => "1"},
    {:id => "2"}
  ]
}

out =
data.each_with_object({}) do |(k,v),o|
    v.chunk_while do |x,y|
        y[:id].to_i == x[:id].to_i + 1
    end
    .each_with_index do |x, i|
        o[i==0 ? "#{k}" : "#{k}_#{i}"] = x
    end
end

pp out
#=>
{"a"=>[{:id=>"1"}, {:id=>"2"}],
 "a_1"=>[{:id=>"5"}, {:id=>"6"}],
 "a_2"=>[{:id=>"9"}, {:id=>"10"}],
 "b"=>[{:id=>"1"}, {:id=>"2"}]}

Comments

[residenti]

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

Answer 2

input = {
  :a => [
    {:id => "1"},
    {:id => "2"},
    {:id => "5"},
    {:id => "6"},
    {:id => "9"},
    {:id => "10"}
  ],
  :b => [
    {:id => "1"},
    {:id => "2"}
  ]
}

pp input.map{|k, v| [k,v.chunk_while{|x, y| y[:id].to_i - x[:id].to_i == 1 }.to_a] }
        .flat_map{|k, v|
          count = -1
          v.map{|l|
            count += 1
            [count == 0 ? "#{k}" : "#{k}_#{count}", l]
          }
        }.to_h

キーがシンボルから文字列に変化していますがそれでいいのでしょうか・・・

というかキーを動的に作成するとプログラムで処理しづらいので配列で対処したほうがよいかと
実際上記プログラムも分割までは1行で終わっていて残りはキーの処理です。

Comments

[residenti]

> キーがシンボルから文字列に変化していますがそれでいいのでしょうか・・・
こちら気付きませんでした。。。修正しました。

また、教えて頂いたコードで無事解決することができました。
ご回答頂きありがとうございましたm

Answer 3

それっぽく動くコードを書いてみました

def create_group_of_id(data)
  ret = {}

  data.keys.each do |key|
    ids = data[key].map { |hash| hash[:id].to_i }
    end_index = 0
    key_count = 0

    while ids[end_index] do
      group = group_continuity_numbers(ids[end_index..-1])
      end_index = ids.index(group.last)
      break if end_index.nil?

      if key_count.zero?
        ret.update(key => group.map { |id| {id: id.to_s} })
      else
        ret.update("#{key}_#{key_count}".to_sym => group.map { |id| {id: id.to_s} })
      end

      key_count += 1
      end_index += 1
    end
  end

  ret
end

def group_continuity_numbers(ids, ret = [])
  return ret if ids.size.zero?

  a, b, *ids = ids

  return ret if a.nil? || b.nil?
  return ret if (a - b).abs >= 2

  group_continuity_numbers([b, ids].flatten, [ret, a, b].flatten.uniq)
end

data =
  {
    :a => [
      {:id => "1"},
      {:id => "2"},
      {:id => "5"}, # <= id が 2から5へと+3増えている。
      {:id => "6"},
      {:id => "9"}, # <= id が 6から9へと+3増えている。
      {:id => "10"}
    ],
    :b => [
      {:id => "1"},
      {:id => "2"},
    ]
  }

pp result = create_group_of_id(data)
# =>
# {
#   :a=>[
#     {:id=>"1"},
#     {:id=>"2"}
#   ],
#  :a_1=>[
#    {:id=>"5"},
#    {:id=>"6"}
#   ],
#  :a_2=>[
#    {:id=>"9"},
#    {:id=>"10"}
#   ],
#  :b=>[
#    {:id=>"1"},
#    {:id=>"2"}
#   ]
# }

他のデータを与えたり詳細なテストはしていないので、もしかしたら動かないパターンもあるかもしれません

なにか参考になれば幸いです

Comments

[residenti]

そっそくご回答頂きありがとうございましたm
こちら無事解決できました。