본문 바로가기
Python/Python_basic

데이터 필터링 예제

by Mr.DonyStark 2023. 11. 15.

open 함수 + r : 파일 불러온 후 조회

□ .readlines() : 데이터를 리스트 형태로 값을 가져옴

□ .rstrip() : 줄바꿈 공백 제거 함수

.split(',') : , 기준으로 분리

.lower() : 소문자로 변환

.startswith('값') : 설정한 값으로 시작하는 것

#필터링 데이터
#예제 : 텍스트 파일을 불러온 후 알파벳 C로 시작하는 나라의 지표 합을 출력하시오

def read_text_file1(file_path):
    #결과 리스트
    value_list = []
    
    with open(file_path, 'r') as file:     #open 함수 + r 읽기/조회
        raw_data = file.readlines()          # .readline() 리스트 형태로 값을 가져옴. raw_data에 리스트형식으로 데이터 적재

        for i in raw_data:
            #.rstrip()을 사용하여 줄바꿈 공백 제거
            #.split(',') : , 기준으로 스플릿하여 country와 value에 값 적재
            country, value = i.rstrip().split(",")   

            if country.lower().startswith('c'):    #country 값 .lower()로 소문자 변환 및 c로 시작하는 값 
                value_list.append(int(value)) #value가 문자열이므로 int 형변환 진행
                
    return value_list

result = read_text_file1('C:/Users/User/Downloads/2.QnA-20231115T061510Z-001/2.QnA/source/41-1.txt')

print(sum(result))