문자열 내 p와 y의 개수

대소문자가 섞여있기 때문에 lower() 함수를 사용했습니다.

if 문으로 'P' or 'p' 와 같이 써도 상관없습니다.

 

for문을 돌며 문자열 요소 하나씩 if문으로 비교합니다.

같이 p, y와 같은지 비교하고 각각 카운트합니다.

 

마지막으로 if문을 사용해 p카운트와 y카운트가 다르면 False값을 가지도록 했습니다.

 

def solution(s):
    answer = True
    s = s.lower()
    p_count = 0
    y_count = 0
    #다르면 false
    for string in s:
        if string is 'p':
            p_count += 1
        if string is 'y':
            y_count += 1
    if p_count is not y_count:
        answer = False

    return answer