Recuperando as primeiras N linhas de um arquivo

Muitas vezes, ao trabalhar com grandes arquivos de dados brutos, torna-se necessário extrair um número específico de linhas para posterior processamento ou análise. Em Python, existem várias abordagens para realizar essa tarefa.

Leitura das primeiras N linhas usando compreensão de lista

Um método simples e eficaz envolve a utilização da compreensão de lista:

with open(path_to_file) as input_file:
    head = [next(input_file) for _ in range(lines_number)]
print(head)

Esta abordagem itera pelo arquivo de entrada usando a função next() e armazena as primeiras linhas_number linhas na lista head.

Usando a função islice()

Outra abordagem aproveita o módulo itertools do Python:

from itertools import islice

with open(path_to_file) as input_file:
    head = list(islice(input_file, lines_number))
print(head)

Aqui, a função islice() é usada para iterar sobre as primeiras linhas_number linhas do arquivo de entrada, criando uma lista das linhas extraídas.

Efeito de Sistema operacional

A implementação descrita acima deve funcionar independentemente do sistema operacional usado. No entanto, é importante notar que no Python 2, a função next() é conhecida como xrange(), o que pode exigir ajustes correspondentes em bases de código mais antigas.