[Python] fileinput 파일 수정하기

안녕하세요.

오늘은 파이썬에서 파일을 입출력하여 수정할 때 사용하는 모듈 fileinput에 대하여 알아보겠습니다.

 

 파이썬 fileinput - 파일 수정

▣ fileinput 모듈

 표준 입력이나 파일 목록에 대한 루프를 빠르게 작성하기 위한 클래스와 함수 모음

 

 - 하나의 파일을 읽거나 수정할 때는 open()을 사용하면 됨

 

▣ 실습 - 파일 수정해보기

: 예시 데이터 (sample_fileinput)

id : 30829
id : 30234
id : 33421
id : 94040
id : 11111
id : 31111

 

 - 텍스트 형식의 예시 데이터를 fileinput 모듈을 활용하여 수정

 - if문을 통해 특정 id의 값을 수정해주고 최종적으로는 sys 모듈의 stdout.write()로 텍스트 파일에 다시 저장해줘야함

import fileinput
import sys
 
# 파일 수정시 inplace = True
for line in fileinput.input('./sample_fileinput.txt', inplace = True):
    if '30829' in line:
        line = line.replace(line, 'name : 30819\n')
    
    # 파일 추가시 sys 활용
    # sys.stdout.write('Text')
    
    # 텍스트 파일 저장하기
    sys.stdout.write(line)

 

: 실행 결과

id : 30819
id : 30234
id : 33421
id : 94040
id : 11111
id : 31111