Exemplo n.º 1
0
 def setUp(self):
     count_trials = 1000
     keys = np.cumsum(np.random.random_integers(1, 10000, count_trials))
     vals = range(count_trials)
     self.indata = zip(keys, vals)
     self.keyfunc = lambda x: x[0]
     self.valfunc = lambda x: x[1]
     self.heap = SlidingMaxHeap(value=self.valfunc)
Exemplo n.º 2
0
 def sliding_max(iterable, window_len, valfunc):
     heap = SlidingMaxHeap(value=valfunc)
     for i, item in enumerate(iterable):
         heap.append(item)
         heap.drop_while(lambda x: len(heap._heap) > window_len)
         yield heap.root