arrays - 将数组拆分为 n、m、o 组或更多组

Question

有什么方法可以像这样拆分数组吗？

[1, 2, 3, 4, 5, 6, 7, 8, 9].split(3, 4, 2)
#=> [[1, 2, 3],[4, 5, 6, 7],[8, 9]]

Answer 1

带有 λ 的不可变版本:

▶ splitter = ->(array, *parts) do 
    parts.reduce([[], 0]) do |acc, i|
      right = acc.last + i
      [acc.first << (acc.last...right), right]
    end.first.map { |r| array[r] }
  end
#⇒ #<Proc:0x0055ae3d9ae7c8@(pry):18 (lambda)>
▶ splitter.((1..9).to_a, 3, 4, 2)
#⇒ [[1, 2, 3], [4, 5, 6, 7], [8, 9]]