본문 바로가기
Coding/초보자를 위한 파이썬 300제

초보자를 위한 파이썬 300제 // 160 split()

by 포스트it 2021. 1. 31.
728x90
반응형

 

문제

파일 이름이 저장된 리스트에서 확장자가 .h나 .c인 파일을 화면에 출력하라.

리스트 = ['intra.h', 'intra.c', 'define.h', 'run.py']
intra.h
intra.c
define.h

내풀이

리스트 = ['intra.h', 'intra.c', 'define.h', 'run.py']
for i in 리스트:
    r = i.split(".")
    if (r[1] == "h") or (r[1] == "c"):
        print(i)
        
결과값
intra.h
intra.c
define.h

정답

논리 연산자 or을 사용해서 두 개의 확장자를 비교합니다.

리스트 = ['intra.h', 'intra.c', 'define.h', 'run.py']
for 변수 in 리스트:
  split = 변수.split(".")
  if (split[1] == "h") or (split[1] == "c"):
    print(변수)

or 는 둘중에 아무거나 상관없이 출력이 되고,
and 는 둘다 만족을해야 출력이 된다.

 

728x90
반응형

댓글