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

문제

 

문자열 s의 길이가 4 혹은 6이고, 숫자로만 구성돼있는지 확인해주는 함수, solution을 완성하세요.

예를 들어 s가 a234이면 False를 리턴하고 1234라면 True를 리턴하면 됩니다.

 

제한 사항

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

 

예시

s	return
a234	false
1234	true

---

풀이

def solution(s):
    answer = False
    if (len(s) == 4 or len(s) == 6) and s.isdigit():
        answer = True

    return answer

 

 

풀이2

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