`Array#each_slice`，在开头留下余数

Question

我正在尝试将一个数组分成三组。我想把余数放在开头（在下面的例子中是[1, 2]）。

arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
...
#=> [[1, 2], [3, 4, 5], [6, 7, 8], [9, 10, 11]]

有什么好办法吗？

拆分数组的常用方法是：

arr.each_slice(3)
# => [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11]]

这会在最后给出余数 [10, 11]。我尝试如下，认为each_slice 可能会接受一个否定论点并将其读作向后遍历数组。

arr.each_slice(-3)

唉，没用。

Answer 1

reverse 和 each_slice 的简单组合就可以解决问题：

arr.reverse.each_slice(3).map(&:reverse).reverse
#=> [[1, 2], [3, 4, 5], [6, 7, 8], [9, 10, 11]]

说明：

我正在反转数组并执行each_slice。那会给我：

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

现在我正在迭代这个并反转每个子数组以匹配预期的子数组序列与.map(&:reverse)。最后，我正在反转整个 Array 以获得所需的序列（最后一个 .reverse）。

Answer 2

虽然可以将数组反转三次，但有更有效的方法来实现目标：

arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
a = arr.dup # not to modify it inplace
[a.shift(a.size % 3)] + a.each_slice(3).to_a
#⇒ [[1, 2], [3, 4, 5], [6, 7, 8], [9, 10, 11]]

顺便说一句，arr.each_slice(3) 返回一个枚举器，而不是您在问题中发布的数组。

更新，或者按照 Cary Swoveland 的建议，取消 dup：

n = a.size % 3
[a[0...n]] + a[n..-1].each_slice(3).to_a

upd 摆脱@sawa 的dup：

[a.first(a.size % 3)] + a.drop(a.size % 3).each_slice(3).to_a

upd 只是出于好奇（假设输入没有 nil 元素）：

([nil] * (3 - a.size % 3) + a).each_slice(3).to_a.tap do |a|
  a.unshift(a.shift.compact!)
end

以上内容可以安全地在原始数组上运行，它不会就地修改它。

UPD2 正如 Stefan 在 cmets 中指出的那样，如果数组可被 3 整除，上述任何操作都将产生一个初始空切片。因此，正确的解决方案（并且，最快的，顺便说一句）应该是这样的：

(arr.size % 3).zero? ? arr.each_slice(3).to_a : ANY_OF_THE_ABOVE