[프로그래머스] level1 : 문자열 다루기 기본

 

파이썬이랑 친해지기

문자열을 다루는데 편리한 기능을 많이 제공한다.

기존에 알고있던 함수로 isalpha()라는 메소드를 사용하여 다음과 같은 코드를 작성했다.

def solution(s):
    for i in s:
        if len(s) != 4 and len(s) != 6:
            return False
        if i.isalpha():
            return False
    return True

위 코드는 문자열 s를 순회하는 과정을 표현하면서 약간은 보기 힘들 수도 있다.

 

파이썬에서 제공하는 isdiget() 함수를 이용해 보다 간결하게 표현할 수 있다.

 

코드

def solution(s):
    return s.isdigit() and len(s) in [4,6]