Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Python에서 파일의 첫 번째 줄만 읽는 방법

파일에서 첫 번째 줄만 읽고 싶다면, 파일을 읽기 모드('r')로 연 후 파일 객체에 readline() 메서드를 호출하면 됩니다. 이 메서드는 파일 포인터를 다음 줄의 시작 위치로 이동시키며, 첫 번째 줄을 문자열로 반환합니다.

readline() 메서드 사용 예제

f = open('my_file.txt', 'r')
line = f.readline()
print(line)
f.close()

위 코드는 my_file.txt 파일에서 첫 번째 줄을 읽어 표준 출력(stdout)으로 출력합니다.

with 문을 활용한 더 안전한 방법

위 방식은 코드 실행 중 예외(exception)가 발생하면 f.close()가 호출되지 않아 파일이 닫히지 않은 채 남을 수 있습니다. 이를 방지하려면 with open 구문을 사용하는 것이 좋습니다. 이 구문은 블록이 종료되거나 예외가 발생해도 파일을 자동으로 닫아주기 때문에 훨씬 안전합니다.

with open('my_file.txt', 'r') as f:
    print(f.readline())

추가 팁: 개행 문자 제거하기

readline()은 줄 끝의 개행 문자(\n)까지 포함하여 반환합니다. 깔끔하게 출력하고 싶다면 strip() 메서드를 함께 사용하세요.

with open('my_file.txt', 'r') as f:
    print(f.readline().strip())

정리하면, 파일의 첫 줄만 읽을 때는 readline() 메서드가 가장 간단한 방법이며, 리소스 관리를 위해 with 문과 함께 사용하는 것이 모범 사례(best practice)입니다.