프로그래머스 - 배열의 원소 삭제하기 Python ver.

 

해결일	언어	레벨	테스트명
20240715	Python	level 0	배열의 원소 삭제하기

 

링크 : https://school.programmers.co.kr/learn/courses/30/lessons/181844

 

 

---

 

내 코드

def solution(arr, delete_list):
    for i in delete_list:
        if i in arr:
            arr.remove(i)
    return arr

 

 

 

개선점:

list에서 remove를 사용하고 싶을 때 해당 아이템이 있으면 삭제 없으면 패스 식으로 처리를 해줄 줄 알았더니, 해당 아이템이 없으면 바로 에러가 떠버리기에... 역시 if가 필요했다.

 

베스트 코드

def solution(arr, delete_list):

    return [i for i in arr if i not in delete_list]