Python: heapq.nlarget, heapq.nsmallest