diff options
Diffstat (limited to 'Lib/heapq.py')
| -rw-r--r-- | Lib/heapq.py | 4 | 
1 files changed, 4 insertions, 0 deletions
| diff --git a/Lib/heapq.py b/Lib/heapq.py index f756035810..dec15aec9c 100644 --- a/Lib/heapq.py +++ b/Lib/heapq.py @@ -185,6 +185,8 @@ def nlargest(n, iterable):      Equivalent to:  sorted(iterable, reverse=True)[:n]      """ +    if n < 0: +        return []      it = iter(iterable)      result = list(islice(it, n))      if not result: @@ -201,6 +203,8 @@ def nsmallest(n, iterable):      Equivalent to:  sorted(iterable)[:n]      """ +    if n < 0: +        return []      if hasattr(iterable, '__len__') and n * 10 <= len(iterable):          # For smaller values of n, the bisect method is faster than a minheap.          # It is also memory efficient, consuming only n elements of space. | 
