У меня есть массив, и я бы хотел выбрать только элементы между двумя указанными значениями.
Например, у меня есть массив, который выглядит так:
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?
Это идеальный пример использования флип-флопа!
a.select {|i| true if (i=="Start")..(i=="Stop")}.slice_before("Start").to_a
Не очень известная функция, но, тем не менее, круто! Используйте его вместе с slice_before, и вы получите именно то, что хотите!
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]}
a.inject [] do |m, e|
if e == 'Start'
@state = true
m << []
end
m.last << e if @state
@state = false if e == 'Stop'
m
end
a.join('^').scan(/Start.*?Stop/).map { |x| x.split('^') }
=> [["Start", "want", "want", "Stop"], ["Start", "want", "Stop"]]
Символ ^ кажется редким, поэтому, возможно, он не будет смешиваться с другими оригинальными символами.
: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"]]