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

문제 설명

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

제한 사항

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

입출력 예

s	return
"a234"	false
"1234"	true

 

풀이

이 문제는 s에 대하여 2가지를 검사해야 한다.

1. s의 길이가 4 또는 6
2. 숫자로만 구성되어 있는지

먼저 s의 길이가 4또는 6인지를 판별하는 코드를 작성한다.

l = len(s)
True if l==4 or l==6 else False

그 다음엔 숫자로 구성되어 있는지 판별해야하는데 파이썬의 String 메서드 중 isdigit()이라는 메서드를 사용하면 문자열이 숫자로만 이루어져 있는지 체크할 수 있다.

True if s.isdigit() else False

위 두개의 조건을 and로 합치면 원하는 답을 구할 수 있다.

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