Аналогичные значения 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]] 

Ive пытался решить эту проблему с помощью group_by однако без успеха.

Благодаря

Answer 1

Я думаю, что это будет делать эту работу, даже если есть, вероятно, является более хорошим решением (как всегда в Ruby):

def group array, diff 
    array.sort! # To remove if you're sure your array will always be sorted 
    # array = array.sort # could be a better solution if you don't want your array to be modified in place 
    res = [] 
    subres = [] 

    array.each do |elt| 
    if subres.count == 0 || elt - subres.last <= diff 
     subres << elt 
    else 
     res << subres 
     subres = [elt] 
    end 
    end 
    res << subres unless subres.empty? 
    res 
end 

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

p group a, 5 
p group a, 2 

выход будет

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

как вы разыскивается.

Answer 2

Использование 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]] 

Answer 3

def group(array, interval) 
    result = [[array[0]]] 
    array.inject do |x, y| 
    if y - x <= interval 
     result.last << y 
     else 
     result << [y] 
    end 
    y 
    end 
    result 
end 

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

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