프로그래머스_python lv0. n의 배수 고르기

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

프로그래머스

 

내 답안 : 

 

def solution(n, numlist):
    return [e for e in numlist if e%n == 0]

 

배수를 구하는 것은 주어진 수로 나누었을때 나머지가 0인 것을 찾으면 된다. 

comprehension 문법으로 한줄로 짧게 배열을 만들었다.

 

 

 

개선안 : 

 

def solution(n, numlist):
    # lambda 식과 filter 함수 사용해 풀이
    # filter(조건 함수, 순회 가능 데이터)    
    return list(filter(lambda v: v%n==0, numlist))

 

아직 람다 식에 익숙하지 않다

filter라는 새로운 함수를 알았다. if문을 사용하지 않고 조건에 맞는 데이터를 뽑아낼 수 있을 것이다.