ruby - Group_by 相似值 - Ruby

Question

我想根据值的接近度和值之间的给定间隔对一个数组中的值进行分组。

a=[1,2,5,7,20,25,50,53]

例如5的间隔:

a=[[1,2,5,7],[20,25],[50,53]]

如果我将间隔更改为 2:

a=[[1,2],[5,7],[20],[25],[50],[53]]

我尝试使用 group_by 来解决这个问题然而没有成功。

谢谢

Answer 1

使用 Enumerable#slice_before :

def group(a, n)
  prev = a.first
  a.slice_before { |x|
    (x - prev > n).tap { prev = x }
  }.to_a
end

a = [1,2,5,7,20,25,50,53]
group(a, 5) # => [[1, 2, 5, 7], [20, 25], [50, 53]]
group(a, 2) # => [[1, 2], [5, 7], [20], [25], [50], [53]]