[Python/프로그래머스] 배열의 원소 삭제하기

Problem

Solution

def solution(arr, delete_list):
    answer = []
    rm_arr = []
    for i in arr:        
        for j in delete_list:
            if i==j:
                rm_arr.append(i)    
                arr = [x for x in arr if x not in rm_arr]                    
    answer = arr
    return answer

Takeaway

리스트명.remove('값') : 리스트 내 하나의 값을 제거, 여러 개의 값을 제거할 수 X
여러 개의 값을 제거하려면? List Comprehension 이용!
ex)