quinta-feira, 9 de outubro de 2014

[Python] - Encontrando os N maiores ou menores termos de uma coleção - Módulo heapq

Olá pessoal, nesse post iremos conhecer o módulo heapq para obter os N maiores ou menores números de uma coleção.

Para isso iremos utilizar dois métodos desse módulo: nlargest e nsmallest.

O código abaixo foi testado nas versões do Python 2.7 e 3.4.


Linha 3: importação do módulo heapq.

Linha 6: criação de uma lista simples de números inteiros.

Linha 8: através da função "nlargest" obtemos os 4 maiores números da lista em questão.

Linha 10: através da função "nsmallest" obtemos os 4 menores números da lista em questão.

O primeiro argumento das funções faladas anteriormente é o "N" e o segundo é a lista.

Linha 13: criação de uma lista mais complexa.

Linha 22: através da função "nlargest" obtemos as 2 pessoas com as maiores idades.

Nas linhas 22 e 24 usamos um parâmetro de chave (key) fazendo com que as funções possam ser utilizadas com estruturas mais complexas.

Linha 24: através da função "nsmallest" obtemos as 2 pessoas com as menores idades.

Quaisquer dúvidas deixem nos comentários, até a próxima.


Nenhum comentário: