Python에서 파일의 처음 N 줄을 추출하는 방법은 무엇입니까?

파일의 첫 N줄 검색

대용량 원시 데이터 파일을 작업할 때 특정 숫자를 추출해야 하는 경우가 종종 있습니다. 추가 처리 또는 분석을 위한 라인. Python에서는 이 작업을 수행하는 데 여러 가지 접근 방식이 있습니다.

목록 이해를 사용하여 첫 N 줄 읽기

간단하고 효과적인 방법은 목록 이해를 활용하는 것입니다.

<code class="python">with open(path_to_file) as input_file:
    head = [next(input_file) for _ in range(lines_number)]
print(head)</code>

이 접근 방식은 next() 함수를 사용하여 입력 파일을 반복하고 헤드 목록에 첫 번째 행_번호 행을 저장합니다.

islice() 함수 사용

또 다른 접근 방식은 Python의 itertools 모듈을 활용합니다.

<code class="python">from itertools import islice

with open(path_to_file) as input_file:
    head = list(islice(input_file, lines_number))
print(head)</code>

여기서 islice() 함수는 입력 파일의 첫 번째 행_번호 행을 반복하여 추출된 행 목록을 생성하는 데 사용됩니다.

운영 체제의 영향

위에 설명된 구현은 사용 중인 운영 체제에 관계없이 작동해야 합니다. 그러나 Python 2에서는 next() 함수가 xrange()로 알려져 있으며 이전 코드 기반에서 해당 조정이 필요할 수 있다는 점은 주목할 가치가 있습니다.

위 내용은 Python에서 파일의 처음 N 줄을 추출하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!