Ruby组非零数字和连续时间

时间:2018-11-06 10:21:02

标签: arrays ruby grouping nmatrix

我有一系列类似这样的项目:[DateTime, value]。我要对满足以下条件的项目进行分组:

  • 3个或更多项目的序列
  • 项目包含的值> 0
  • 时间是连续的(增加1秒)

我想获取满足这些条件的序列的开始和结束的索引。

例如

[ 
[today 10:00:00, 1],
[today 10:00:01, 1],
[today 10:00:02, 1],
[today 10:00:03, 0],
[today 10:00:04, 1],
[today 10:00:05, 1],
[today 10:00:16, 1],
[today 10:00:17, 1],
]

应返回:

[ [0,2] ]

我看过DaruNMatrix,但我不知道如何进行顺序比较来做我想做的事情。

现在我有一个很大的循环,可以进行很多比较-有更好的方法吗?

2 个答案:

答案 0 :(得分:2)

t = Time.now
a = [
  [t + 0, 1],
  [t + 1, 1],
  [t + 2, 1],
  [t + 3, 0],
  [t + 4, 1],
  [t + 5, 1],
  [t + 16, 1],
  [t + 17, 1],
]

a
.each_with_index
.chunk_while{
  |(x, i), (y, j)|
  x[1].positive? and
  y[1].positive? and
  x[0].to_i.next == y[0].to_i
}
.select{|chunk| chunk.length >= 3}
.map{|chunk| [chunk.first[1], chunk.last[1]]}
# => [[0, 2]]

答案 1 :(得分:1)

t = Time.now
arr = [[t+ 0, 1], [t+ 1, 1], [t+ 2, 1], [t+ 3, 0], [t+ 4, 1], [t+ 5, 1], [t+16, 1],
       [t+18, 1], [t+19, 1], [t+20, 1], [t+21, 1], [t+30, 1]]
  #=> [[2018-11-06 10:11:52 -0800, 1], [2018-11-06 10:11:53 -0800, 1],...,
  #    [2018-11-06 10:12:22 -0800, 1]]

arr.each_index.
    slice_when { |i,j| arr[i].last.zero? || arr[i-1].last.zero? ||
      (arr[i].first - arr[i-1].first > 1) }.
    each_with_object([]) { |a,b| b << [a.first, a.last] if a.last-a.first >= 2 }
  #=> [[0, 2], [7, 10]]

Emumerable#slice_when(MRI v2.2的新功能)与Enumerable#chunk_while(MRI v2.3的新功能)密切相关,@ sawa在回答中使用了它。通常,如果可以使用一个,则可以选择另一个。

要支持早期版本的Ruby,可以使用Enumerable#slice_before(MRI v1.9.2中的新功能)。

arr.each_index.
    slice_before { |i| i > 0 &&
      (arr[i].last.zero? || arr[i-1].last.zero? || (arr[i].first - arr[i-1].first > 1)) }.
    each_with_object([]) { |a,b| b << [a.first, a.last] if a.last-a.first >= 2 }

注意中间计算:

enum = arr.each_index.slice_before {|i| i > 0 &&
  (arr[i].last.zero? || arr[i-1].last.zero? || (arr[i].first - arr[i-1].first > 1)) }
  #=> => #<Enumerator: #<Enumerator::Generator:0x0000000001948d50>:each>
enum.to_a
  #=> [[0, 1, 2], [3], [4, 5], [6], [7, 8, 9, 10], [11]]

(使用slice_when时返回值是相同的。)如果需要,该子句

each_with_object([]) { |a,b| b << [a.first, a.last] if a.last-a.first >= 2 }

可以用两遍替代方法代替:

select { |a| a.last - a.first >= 2 }.map { |a| [a.first, a.last] }