Извлечение первых N строк файла

Часто при работе с большими файлами необработанных данных возникает необходимость извлечь определенное число строк для дальнейшей обработки или анализа. В Python существует несколько подходов к выполнению этой задачи.

Чтение первых N строк с использованием понимания списка

Простой и эффективный метод включает использование понимания списка:

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

Этот подход перебирает входной файл с помощью функции next() и сохраняет первые строкиlines_number в списке заголовков.

Использование функции islice()

Другой подход использует модуль Python itertools:

from itertools import islice

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

Здесь функция islice() используется для перебора первых строк_номер_строки входного файла, создавая список извлеченных строк.

Эффект Операционная система

Описанная выше реализация должна работать независимо от используемой операционной системы. Однако стоит отметить, что в Python 2 функция next() известна как xrange(), что может потребовать соответствующих изменений в старых базах кода.