use heapq.nlargest for fixed-size sort