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

 

문제

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

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

내풀이

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

정답

split() 메서드로 문자열을 분할하고 확장자가 "h"인지 분기문으로 비교합니다. 확장자가 "h"라면 파일 이름을 출력합니다.

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

---

split 과 if문을 잘 활용한 문제입니다.