如何按索引对数组元素进行分组？

Question

我有数组arr，我想按数组idx 中给出的索引对其进行分组。我的意思是，

• 子数组 1 将在索引 1 处结束
• 子数组 2 将在索引 5 处结束
• 子数组 3 将在索引 7 处结束
• 子数组 N 将从索引 8 处的元素到最后一个元素形成 arr

使用我当前的代码，我可以将第一个索引为idx idx[0] = 1 的子数组分组。

那么，如何复制数组idx 中的所有索引？提前致谢。

我当前的代码和输出是这样的：

idx = [1,5,7]
arr = ['a','b','c','d','e','f','g','h','i','j','k']

arr.group_by.with_index { |z, i| i <= idx[0] }.values
=> [["a", "b"], ["c", "d", "e", "f", "g", "h", "i", "j", "k"]]

我想要的输出是这样的：

output   --> [["a", "b"], ["c", "d", "e", "f"], ["g", "h"], ["i", "j", "k"]]

#Indexes -->    0    1      2    3    4    5      6    7      8    9    10  

Answer 1

您可以使用slice_after 在索引位于idx 中的每个项目之后对数组进行切片：

idx = [1, 5, 7]
arr = %w[a b c d e f g h i j k]

arr.enum_for(:slice_after).with_index { |_, i| idx.include?(i) }.to_a
#=> [["a", "b"], ["c", "d", "e", "f"], ["g", "h"], ["i", "j", "k"]]

enum_for（不幸的是）需要链接slice_after 和with_index。

Answer 2

另一种解决方案

idx = [1, 5, 7]
arr = ['a','b','c','d','e','f','g','h','i','j','k']
from = 0

arr.map.with_index { |a, i|  
  if idx.include?(i)
    result = arr[from..i]  
    from = i + 1 
  end
  result
}.compact

 => [["a", "b"], ["c", "d", "e", "f"], ["g", "h"]] 

Answer 3

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

[-1, *end_idx, arr.size-1].uniq.each_cons(2).
  map { |s,e| arr.values_at(s+1..e) }
  #=> [[9, 3], [1, 6, 2, 4], [0, 1], [5, 8]]  

步骤如下：

a = [-1, *end_idx, arr.size-1]
  #=> [-1, 1, 5, 7, 9]
b = a.uniq
  #=> [-1, 1, 5, 9]
c = b.each_cons(2)
  #=> #<Enumerator: [-1, 1, 5, 7, 9]:each_cons(2)> 
c.map { |s,e| arr.values_at(s+1..e) }
  #=> [[9, 3], [1, 6, 2, 4], [0, 1], [5, 8]] 

通过将c 转换为数组，可以看到枚举器c 生成并传递给map 的元素。

c.to_a
  #=> [[-1, 1], [1, 5], [5, 7], [7, 9]] 

见Array#values_at。