[프로그래머스/programmers] 문자열 내 p와 y의 개수

def solution(s):
    pnum = 0
    ynum = 0

    for i in s:
        if i == "p" or i == "P":
            pnum += 1
        elif i == "y" or i == "Y":
            ynum += 1
    if pnum == ynum:
        answer = True
    elif pnum != ynum:
        answer = False
    else:
        answer = True
    return answer

=>

오,,, lower

count 함수 사용하기

 

new code