rubyで配列を任意のindex番号で分割する方法を教えてください

Question

ruby(railsでも可)で下記のような配列を任意のindex番号で分割する方法を教えてください。

arr = [1,2,3,4,5,6,7,8,9,10]

の配列を、任意のindex番号が2, 4, 7だった場合に下記のように分けたいです。
(指定したindex番号が先頭になる配列に分割したい)

[1,2]
[3,4]
[5,6,7]
[8,9,10]

今考えているのは

arr = [1,2,3,4,5,6,7,8,9,10]
offsets = [2, 4, 7]
offset = 0

result = offsets.each_with_object([]) do |n, result|
  result.push arr[offset..(n - 2)]
  offset = n - 1
end
result.push arr[offset..-1]

という感じです。offsetsは数値を直書きしてますが実際は要素数が不定、かつindex番号も不定の変数になります。(index番号は元の配列の長さを超えないこととします)
この場合、どのように記述すれば↑のコードより完結かつ高速に処理できるか教えていただきたいです。

何卒よろしくお願いします！

Comments

[winterboum]

indexの先頭は１ですか？
rubyの場合は０が普通なので

[usestrict]

失礼しました、例文が間違っておりました！

Answer 1

div.rb

arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
offsets = [2, 4, 7]

arr_2 = ([0] + offsets + [0])
          .each_cons(2).map{|i, j| arr[i .. j - 1]}
p(arr_2)

実行例

Comments

[winterboum]

これだ！！
確かそんなmethodがあったはずだと思い出そうとしたのですが、思い出しきれませんでした。

Answer 2

arr = [1,2,3,4,5,6,7,8,9,10]
offsets = [2, 4, 7]

offsets << arr.size
p arr.chunk.with_index{|_,i| offsets.bsearch{|x| x > i} }.map(&:last)

正直、初めてArray#bsearch / [Enumerable#chunk](https://docs.ruby lang.org/ja/latest/method/Enumerable/i/chunk.html)を使ったような気がします。

追記

arr = [1,2,3,4,5,6,7,8,9,10]
offsets = [2, 4, 7]

p arr.each_with_index.slice_when{|_, (_, i)|
  if offsets.first == i
    offsets.shift
  end
}.map{|a| a.map(&:first)}

オフセット→要素数への変換ができるのならば

arr = [1,2,3,4,5,6,7,8,9,10]
elements = [2, 2, 3]
result = []
result << arr.shift(elements.shift) until elements.empty?
result << arr
p result

が速いとは思います。

Comments

[otn]

> オフセット→要素数への変換ができるのならば
それは、この人の1つ前の質問です。
個数だと、arrが余った場合にどうしたいのかが不明ですね。

[winterboum]

そういうchunkの使い方がありましたか

Answer 3

rubyの場合indexは０から扱うほうが何かと楽なので
「任意のindex番号が3,5,8だった場合に」
を
「任意のindex番号が2,4,7だった場合に」
とします。

arr = [1,2,3,4,5,6,7,8,9,10]
indexes = [2,4,7]
w_indexes = [0]+indexes
d_indexes = indexes.map.with_index{|i,idx| i - work[idx]}
darr = arr.dup
d_indexes.map{|i| darr.shift(i)}+[darr]

ん〜〜ちと長いな。

arr = [1,2,3,4,5,6,7,8,9,10]
indexes = [2,4,7]
k=0;ary=[];
darr=arr.dup
indexes.each{|idx| ary << darr.shift(idx-k);k=idx};ary<<darr

似たようなものか

Comments

[usestrict]

失礼しました！例文が間違っていたため修正しました！

[winterboum]

それに沿っていると思います