Как найти интервалы непустых значений в Ruby NArray? - PullRequest
Новая
       43

Как найти интервалы непустых значений в Ruby NArray?

1 голос
/ 19 октября 2011

Мне интересно найти все непустые интервалы в NArray как кортежи типа [begin, end].Поэтому, если у нас есть NArray заданного размера и все значения в позициях индекса 100 ... 200, 300 ... 400 и т. Д. Отличны от нуля, я хотел бы получить массив, подобный этому: [[100,200], [300,400], etc].Интересно, есть ли что-то в интерфейсе NArray - что, к сожалению, недостаточно документировано - что может мне помочь?Скорость имеет решающее значение.

Приветствия,

Мартин

1 Ответ

1 голос
/ 20 октября 2011 
$ irb -rnarray
irb> a = NArray[0,1,2,0,0,5,0,7,8,9]
irb> tmp = NArray.new( a.typecode, a.size+2 )
irb> tmp[1..-2] = a
irb> tmp
=> NArray.int(12): 
[ 0, 0, 1, 2, 0, 0, 5, 0, 7, 8, 9, 0 ]

irb> empty = tmp.eq(0)
=> NArray.byte(12): 
[ 1, 1, 0, 0, 1, 1, 0, 1, 0, 0, 0, 1 ]

irb> beg = (~empty[1..-1] & empty[0..-2]).where
=> NArray.int(3): 
[ 1, 5, 7 ]

irb> fin = (empty[1..-1] & ~empty[0..-2]).where
=> NArray.int(3): 
[ 3, 6, 10 ]

irb> range = NArray[beg,fin].transpose(1,0)
=> NArray.int(2,3): 
[ [ 1, 3 ], 
  [ 5, 6 ], 
  [ 7, 10 ] ]
...