Python에서 파일의 첫 N 줄 읽기

데이터 처리에서는 대규모 데이터 세트의 일부만 조작해야 하는 경우가 많습니다. . 여기서는 텍스트 파일에서 지정된 수의 줄을 읽는 기능이 작동하는 곳입니다.

Python의 내장 메서드

Python은 읽기 쉬운 방법을 제공합니다. 파일의 첫 번째 N 줄:

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

이 코드 조각은 지정된 경로에서 파일을 연 다음 라인을 라인 수만큼 반복하여 검색된 라인을 헤드 목록에 저장합니다.

운영 체제 고려 사항

운영 체제는 일반적으로 이 작업의 구현에 영향을 미치지 않습니다. Python은 기본 시스템에 관계없이 파일 액세스를 관리하므로 다양한 플랫폼 간에 코드를 이식할 수 있습니다.

islice를 사용한 대체 접근 방식

첫 번째 N 줄을 읽는 또 다른 옵션은 다음과 같습니다. itertools 모듈의 islice 함수를 사용하십시오:

from itertools import islice

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

이 메소드는 파일의 첫 번째 행_line_number 행을 생성하는 생성기 반복자를 리턴합니다. list() 함수는 쉽게 처리할 수 있도록 생성기를 목록으로 변환하는 데 사용됩니다.

결론

이 코드 조각은 지정된 행 수를 읽는 안정적인 방법을 제공합니다. Python의 텍스트 파일에서. 대규모 데이터 세트를 트리밍하든 처음 몇 줄에서 특정 작업을 수행하든 이러한 방법은 효율적이고 다양한 솔루션을 제공합니다.