nb_heapq

Functions

heap_push(heap, size, element)	Heapq heappush
heap_pop(heap, size)	Heapq heappop
init_heap([num_rows, num_compare])	Initialise heap with (num_compare + 2) * num_rows elements, where

Module Contents

nb_heapq.heap_push(heap, size, element)

Heapq heappush

nb_heapq.heap_pop(heap, size)

Heapq heappop

nb_heapq.init_heap(num_rows=4, num_compare=1)

Initialise heap with (num_compare + 2) * num_rows elements, where num_compare is the number of elements to order by in lexicographical order, and the remaining two elements are the file and row idxs

Currently limited to dtype np.int32 as it is tricky to use custom dtypes