对数组的每个第 n 个元素进行分组

Question

我有一个数组：

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

如何对数组的每 n 个元素进行分组？例如，对于 3:

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

这段代码是我写的，但是很丑：

array.each_with_index.group_by { |e, i| i % 3}.map {|h| h[1].map { |e| e[0] }}

Answer 1

试试这个方法：

irb(main):002:0> a =  Array(1..9)
=> [1, 2, 3, 4, 5, 6, 7, 8, 9]
irb(main):004:0> a.each_slice(3).to_a.transpose # good
=> [[1, 4, 7], [2, 5, 8], [3, 6, 9]]

不是你想要的？让我知道

Answer 2

这是一个简单的方法：

array.each_slice(3).to_a.transpose
#=> [[1, 4, 7], [2, 5, 8], [3, 6, 9]]

Answer 3

Nick Veys 的回答最为直截了当，但这里有另一种方式。

array.group_by.with_index{|_, i| i % 3}.values
#=> [[1, 4, 7], [2, 5, 8], [3, 6, 9]]

Answer 4

这也可以通过创建散列，然后提取值来完成。对于 Ruby 1.9+，这将保留数组元素的顺序。

代码

def bunch(arr,ngroups)
  arr.each_with_index.with_object(Hash.new { |h,k| h[k]=[] }) { |(e,i),h|
       h[i%ngroups] << e }
     .values
end

例子

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

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

备选方案

如果参数是每个组的大小，grp_size，我们可以计算：

ngroups = (arr.size+grp_size-1)/grp_size

该方法的操作行可以改写为：

arr.each_with_index.with_object({}) { |(e,i),h| (h[i%ngroups] ||= []) << e }
   .values

对数组的每个第 n 个元素进行分组

Group every n-th element of array

ruby

arrays

grouping