В стандартной библиотеке есть модуль
heapq, который, как можно понять по названию, реализует структуру даных куча.Особенности и пользу этой структуры данных пока что разбирать не будем, но посмотрим на два интересных метода в этом модуле.
С помощью методов
nsmallest и nlargest можно получить наименьшие и наибольшие элементы в списке. Первым аргументов передается, сколько элементов нужно взять, вторым — список.
#heap