Групповой массив Ruby по определенным значениям

У меня есть массив, и я бы хотел выбрать только элементы между двумя указанными значениями.

Например, у меня есть массив, который выглядит так:

a = ["don't want", "don't want", "Start", "want", "want", "Stop", "don't want", "Start", "want", "Stop", "don't want"]

Я хочу вызвать метод в массиве, который захватывает элементы между "Start" и "Stop" (включая элементы "Start" и "Stop" ). Результирующий массив должен выглядеть следующим образом:

[["Start", "want", "want", "Stop"], ["Start", "want", "Stop"]]

Я не мог найти такой метод, поэтому я попытался написать один:

class Array
  def group_by_start_and_stop(start = "Start", stop = "Stop")
    main_array = []
    sub_array = []

    group_this_element = false

    each do |e|
      group_this_element = true if e == start

      sub_array << e if group_this_element

      if group_this_element and e == stop
        main_array << sub_array
        sub_array = []
        group_this_element = false
      end
    end

    main_array
  end
end

Это работает, но, похоже, это, возможно, излишне многословно для меня. Поэтому у меня есть два вопроса: существует ли аналогичный метод? Если нет, можно ли улучшить метод group_by_start_and_stop?

Ответ 1

Это идеальный пример использования флип-флопа!

a.select {|i| true if (i=="Start")..(i=="Stop")}.slice_before("Start").to_a

Не очень известная функция, но, тем не менее, круто! Используйте его вместе с slice_before, и вы получите именно то, что хотите!

Ответ 2

a.each_with_index.select{|s, i| s.eql?("Start") or s.eql?("Stop")}
                 .map{|s, i| i}
                 .each_slice(2)
                 .map{|s, f| a[s..f]}

Ответ 3

a.inject [] do |m, e|
  if e == 'Start'
    @state = true
    m << []
  end
  m.last << e if @state
  @state = false if e == 'Stop'
  m
end

Ответ 4

a.join('^').scan(/Start.*?Stop/).map { |x| x.split('^') }
=> [["Start", "want", "want", "Stop"], ["Start", "want", "Stop"]]

Символ ^ кажется редким, поэтому, возможно, он не будет смешиваться с другими оригинальными символами.

Ответ 5

:dummy ниже - убедиться, что первый фрагмент возврата из slice_before - это то, что вы не хотите.

p [pres = :dummy, *a].slice_before{|e|
  prev, pres = pres, e
  prev == "Stop" or pres == "Start"
}
.select.with_index{|_, i| i.odd?}
# => => [["Start", "want", "want", "Stop"], ["Start", "want", "Stop"]]