[프로그래머스] 머쓱이보다 키 큰 사람 (Python)

문제

문제링크

친구들의 키가 담긴 정수 배열 array와 머쓱이의 키 height가 주어질 때,

머쓱이보다 키 큰 사람 수를 return하도록 solution 함수를 완성하라.

 

풀이

리스트 컴프리헨션과 len()을 조합해 리스트에 height보다 큰 요소만 남기고 남은 요소들의 개수를 센다.

 

코드

def solution(array, height):
    answer=len([i for i in array if i>height])
    return answer