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

Материал из Онлайн справочника
Перейти к навигацииПерейти к поиску

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


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

Задача

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

Решение

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

Пример

№1

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

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

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

import heapq

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

print(heapq.nlargest(3, nums))

Результат:

[111, 100, 99]

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

print(heapq.nsmallest(3, nums))

Результат:

[-38, -9, -4]

№2

Разберем пример с более сложной структурой данных, дан список отображений:

crypto= [
	{'name':'bitcoin','price': 7500, 'amount':50},
	{'name':'litecoin','price': 75, 'amount':75},
	{'name':'ethereum','price': 750, 'amount':115},
	{'name':'dogecoin','price': 0.075, 'amount':20000},
	{'name':'tron','price': 0.75, 'amount':1750},
	{'name':'monero','price': 65.75, 'amount':100}]

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

import heapq

Получим 3 отображения с максимальной ценой:

print(heapq.nlargest(3, crypto, key=lambda s: s['price']))

Результат:

[{'name': 'bitcoin', 'price': 7500, 'amount': 50}, {'name': 'ethereum', 'price': 750, 'amount': 115}, {'name': 'litecoin', 'price': 75, 'amount': 75}]

Получим 3 отображения с минимальным количеством монет:

print(heapq.nsmallest(3, crypto, key=lambda s: s['amount']))

Результат:

[{'name': 'bitcoin', 'price': 7500, 'amount': 50}, {'name': 'litecoin', 'price': 75, 'amount': 75}, {'name': 'monero', 'price': 65.75, 'amount': 100}]

См.также

Внешние ссылки