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

정규표현식을 사용하면 진짜 짧게 풀 수 있을거 같은데

 

나는 허접이라 남이 쓰던 코드 findall 가져와서 썻다..

 

근데 정규표현식을 공부해서 다시 풀고 싶다

 

def solution(s):
    import re
    if len(s) == 4 or len(s) == 6:
        s = re.findall('\d+', s)[0]
        if len(s) == 4 or len(s) == 6:
            return True
    return False