[프로그래머스] 나누어 떨어지는 숫자 배열(Python)

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/12910

풀이

풀이 설명

[출처] https://wikidocs.net/22803

• filter(): 특정 조건으로 걸러, 걸러진 요소들로 iterator객체를 만들어서 리턴해주는 함수.
  • filter를 사용할 땐, list로 다시 감싸줘야 편하다.

    answer = list(filter(lambda x : x < 5, arr))

코드

def solution(arr, divisor):
    answer = []

    answer = list(filter(lambda x: x % divisor == 0, arr))

    if not answer:
        answer.append(-1)
    else:
        answer.sort()

    return answer

 ##참고 코드
 # return ( sorted([filter(lambda x: x % divisor == 0, arr)] or [-1]) )
 # return 에서 empty list는 어차피 False로 인식하기 때문에, or 을 이용하여 [-1] 로 대입하게 한다.