按特定值分组Ruby数组

时间:2012-11-10 19:12:30

标签: ruby arrays

我有一个数组,我只想选择两个指定值之间的元素。

例如,我有一个如下所示的数组:

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

我想在一个数组上调用一个方法来捕获“开始”和“停止”之间的元素(包括“开始”和“停止”元素)。生成的数组应如下所示:

[["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方法更好?

5 个答案:

答案 0 :(得分:7)

这是触发器有用的完美示例!

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

不是超级知名的功能,但还是很酷!与slice_before一起使用,就可以得到你想要的东西!

答案 1 :(得分:1)

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]}

答案 2 :(得分:1)

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

答案 3 :(得分:0)

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

符号 ^ 似乎很少见,所以它可能不会与其他原始符号混合。

答案 4 :(得分:0)

下面的

: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"]]