정수 n과 정수 배열 numlist가 매개변수로 주어질 때, numlist에서 n의 배수가 아닌 수들을 제거한 배열을 return하도록 solution 함수를 완성해주세요.
입출력 예 #1
numlist에서 3의 배수만을 남긴 [6, 9, 12]를 return합니다.
입출력 예 #2
numlist에서 5의 배수만을 남긴 [10, 5]를 return합니다.
입출력 예 #3
numlist에서 12의 배수만을 남긴 [120, 600, 12, 12]를 return합니다.
# 풀이 방법 1
def solution(n, numlist):
result = []
for i in numlist:
if i % n == 0:
result.append(i)
return result
# 풀이 방법 2
def solution(n, numlist):
result = [num for num in numlist if num % n == 0]
return result
리스트 컴프리헨션(list comprehension)을 사용했다.
num을 numlist 배열 안의 숫자라고 했을 때,
n의 배수를 구하는 것이므로 for in 반복문을 돌려서 num을 n으로 나눴을 때 나머지가 0인 요소를 포함하는 새로운 배열을 만들었다.