파일의 첫 번째 N 줄 검색

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

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

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

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

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

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()로 알려져 있으며 이전 코드 베이스에서 해당 조정이 필요할 수 있다는 점에 주목할 가치가 있습니다.