Receta Python No. 1-4: Encontrar los N Elementos Mínimos y Máximos

1. Problema

Encontrar el conjunto de elementos N de valores mínimos y máximos de una colección.

2. Solución

El módulo Python headq cuenta con dos funciones interesantes para encontrar los N elementos mínimos y máximos: nsmallest() y nlargest() (Beazly, Jones, 2013).

3. Código Python

Un ejemplo básico podría ser:

import heapq

numeros = [43, 13, 29, -2, -4, 11, 53, 17, 47, 61, -11]

print(heapq.nsmalest(3, numeros))
print(heapq.nlargest(3, numeros))

Prueba de ejecución:

[-11, -4, -2]

[61, 53, 47]
Otro ejemplo que involucra una estructura de datos más compleja:

Archivo Python productos-baratos-caros.py [enlace alternativo]:

Prueba de ejecución:

[{'nombre': 'Disco blue-ray', 'precio': 7.5, 'marca': 'Verbatim'}, {'nombre': 'Memoria USB', 'precio': 13.9, 'marca': 'Kingston'}]

[{'nombre': 'Teclado', 'precio': 42.7, 'marca': 'Logitech'}, {'nombre': 'Audífonos', 'precio': 35.5, 'marca': 'Microsoft'}]

4. Literatura & Enlaces

Beazly D, Jones B. K. (2013). Python Cookbook, Third Edition. United States: O'Reilly Media.

O