파이썬 (41) 텍스트 파일 필터링

이번에 수행하는 예제는 외부 파일을 읽어오고

읽어온 파일에서 대문자 C로 시작하는 행의 2번째 열을 전부 합한 값을 반환하는 것이다

Russia,17075200
Canada,9984670
United States of America,9826630
China,1385566537
Brazil,200361925
Australia,23342553
India,1252139596
Argentina,41446246
Kazakhstan,16440586
Algeria,39208194
.
.
.

 

파일은 위와 같이 구성되어 있다

 

- 스크립트

import os

def readText(filepath):
    ret = 0
    if os.path.exists(filepath):
        with open(f"{filepath}", 'r') as f:
            txt = f.read().split("\n")
            for i in txt:
                if i.startswith("C") == True:
                    ret += int(i.split(",")[1])

    return ret

print(readText("Target.txt"))

 

함수 readText를 통해 해당 작업을 수행한다

우선, os 모듈의 exists 함수를 통해 인자로 받은 경로에 파일이 존재하는지 확인한다

 

읽어온파일을 split 함수를 통해 개행을 제거하고 하나의 리스트에 전부 저장한다

리스트에 저장된 파일을 순회하면서 startswith 함수를 통해 판단하여 대문자 C로 시작한다면

순회중인 대상을 ,(쉼표)로 분리한 후에 뒤에 위치한 숫자 문자열을 정수형으로 바꿔 값을 더한다

 

최종적으로 전부 더해진 값을 반환한다

최종 실행 결과는 다음과 같다