프로그래머스 LV.1 - 문자열 다루기 기본[Python]

문제 설명

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

제한 사항

• s는 길이 1 이상, 길이 8 이하인 문자열입니다.
• s는 영문 알파벳 대소문자 또는 0부터 9까지 숫자로 이루어져 있습니다.

입출력 예sreturn

"a234"	false
"1234"	true

 

<풀이>

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

    else:
        return False

isdigit을 통해 숫자인지 확인해주고 len(s)의 길이가 (4, 6)에 포함되는지 확인해주는 코드입니다.

 

<한줄 코딩>

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