Python:Рецепты/Поиск N максимальных и минимальных элементов

Содержание

Резерв

Рецепты

Примеры

Библиотека

Резерв

Резерв

Резерв

Резерв

Резерв

Проверка/Оформление/Редактирование: Мякишев Е.А.

Поиск N максимальных и минимальных элементов

Задача

У вас есть коллекция элементов и вы хотите создать список из N максимальных или минимальных элементов.

Решение

Для этого подойдёт модуль heapq, в котором есть две функции nlargest() и nsmallest().

Пример

№1

Дана последовательность чисел:

nums = [111, -4, 0, 10, 56, -38, 11, 78, -9, 100, 99, 1,2,3]

Импортируем модуль:

import heap q

Получим 3 максимальных числа из последовательности:

print(heapq.nlargest(3, nums))

Результат:

[111, 100, 99]

Получим 3 минимальных числа из последовательности:

print(heapq.nsmallest(3, nums))

Результат:

[-38, -9, -4]