[프로그래머스 / Python] 문자열 다루기 기본 | 초코더

문제 설명

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

 

제한 사항

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

 

입출력 예

s	return
a234	false
1234	true

 

접근법

문자열의 길이가 4또는6인지 확인해준 다음, 숫자인지 문자가 섞였는지 확인해줍니다.

 

처음에는 문자열의 각각 원소를 list에 넣어준 후 각각이 숫자인지 문자인지 비교해주려 했습니다.

예) a234 -> ["a","2","3","4"] -> for value in s :

 

하지만 isdigit()함수를 통해 s.isdigit()을 하면 숫자인지 숫자가 아닌지 판별할 수 있었습니다.

isalpha()는 문자열인지 확인해주는 함수.

​

나의 코드

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