본문 바로가기

Python

[Python/level 0] n의 배수 고르기 - 120905, 리스트 컴프리헨션

문제 설명

정수 n과 정수 배열 numlist가 매개변수로 주어질 때, numlist에서 n의 배수가 아닌 수들을 제거한 배열을 return하도록 solution 함수를 완성해주세요.

 

code - 1차

def solution(n, numlist):
    answer = []
    for i in numlist:
        if i%n == 0:
            answer.append(i)
    return answer

 

code - 2차

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

 

반응형