Programmers with Python [Level1] - 문자열 다루기 기본

문제 설명

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

• 제한 사항

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

• 입출력 예시
  

코드

# https://school.programmers.co.kr/learn/courses/30/lessons/12918
# programmers, level1 - 문자열 다루기 기본, python3
def solution(s):
    if len(s) != 4 and len(s) != 6:
        return False
    for i in s:
        if i.isalpha():
            return False
    return True

# or

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

결과