Python

[Python] 리스트 특정 원소 모두 지우기

KEMON 2020. 9. 13. 22:17
728x90

아래와 같이 파이썬의 리스트에서 remove()는 한 가지만 지우게 된다.

a = [1, 2, 3, 3]
a.remove(3)  #위의 배열에 3이 2개있지만 하나만 제거됨
print(a)    # 결과 값 : [1, 2, 3]

따라서 리스트에서 특정 원소를 모두 지우는 방법은 아래와 같다.

a = [1, 2, 3, 4, 5, 5, 5]
remove_set = [3, 5] #지우고 싶은 원소

#remove_set에 포함되지 않은 값만 저장
result = [i for i in a if i not in remove_set] 
print(result) #결과 값 : [1, 2, 4]

 

728x90