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

문제 설명

문자열 s의 길이가 4 혹은 6이고, 숫자로만 구성돼있는지 확인해주는 함수, solution을 완성하세요. 예를 들어 s가 "a234"이면 False를 리턴하고 "1234"라면 True를 리턴하면 됩니다.

 

제한 사항

• s는 길이 1 이상, 길이 8 이하인 문자열입니다.

 

코딩테스트 연습 - 문자열 다루기 기본

 

풀이

def solution(s):
    if (len(s) == 4 or len(s) == 6):
        try:
            int(s)
            return True
        except:
            return False
    else:
        return False

 

다른풀이

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

isdigit()을 통해 문자열이 숫자로 이루어져 있는지 판단할 수 있다.

번외로 isalpha()를 통해 문자열이 문자로 이루어져 있는지 판단할 수 있다.