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

时间:2015-02-06 17:52:55

标签: arrays ruby grouping

我有一个数组:

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] }}

4 个答案:

答案 0 :(得分:5)

这是一个简单的方法:

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

答案 1 :(得分:4)

Nick Veys的回答是最简单的,但这是另一种方式。

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

答案 2 :(得分:0)

试试这个:

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]]

不是你想要的?让我知道

答案 3 :(得分:0)

这也可以通过创建哈希,然后提取值来完成。对于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