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

in Algorithm on Programmers, Problems


문제

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

문제 설명 및 풀이

isdigit으로 숫자인지 쉽게 확인할 수 있다.

C++스러운 나의 코드

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

    return False

Pythonic한 풀이

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