我有一系列类似这样的项目:[DateTime, value]
。我要对满足以下条件的项目进行分组:
我想获取满足这些条件的序列的开始和结束的索引。
例如
[
[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] ]
我看过Daru和NMatrix,但我不知道如何进行顺序比较来做我想做的事情。
现在我有一个很大的循环,可以进行很多比较-有更好的方法吗?
答案 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] }