[Python] split 함수

split 함수를 쓰면 리스트로 담긴다

name = 'first.middle.last'

#구분자 없이
call = name.split()
print(call)

#구분자 포함
call = name.split('.')
print(call)

>>  ['first.middle.last']

>>  ['first','middle','last']

 

리스트로 담기기 때문에 원하는 것만 따로 불러낼 수 있음

 

Ex) BOJ/백준 20291 - 파일 정리

 

여기서 뒤에 파일 확장자명만 필요하다. split 함수를 이용해 확장자명만 바로 받아보자.

array = []
for _ in range(int(input())):
    extens = input().split('.')[1]
    array.append(extens)
print(array)

>>  ['txt','spc','icpc','icpc','txt','world','spc','txt']