수학 시간에 배우는 '집합(set)' 개념이라면 누구나 익숙할 것입니다. 파이썬의 set 자료형은 수학의 집합과 동일한 개념을 구현한 데이터 구조로, 다양한 요소들을 담을 수 있으며 요소 간에는 순서가 정의되지 않습니다. 세트에서는 요소를 자유롭게 추가하거나 삭제할 수 있고, 반복문으로 요소를 순회할 수 있으며, 합집합·교집합·차집합과 같은 표준 집합 연산도 손쉽게 수행할 수 있습니다.
이 글에서는 주어진 세트에서 요소를 하나씩 제거하는 방법을 살펴봅니다. 여기서는 파이썬 내장 메서드인 pop()을 사용합니다. pop() 메서드는 세트에서 요소를 하나씩 꺼내어(제거하고) 그 값을 반환하는 기능을 합니다. 세트는 순서가 없는 자료형이므로, pop() 호출 시 어떤 요소가 제거될지는 정해져 있지 않다는 점을 기억하세요.
예시
A = [2, 9, 80, 115, 22, 120]
출력 결과
NEW SET IS ::>
{9, 80, 115, 22, 120}
{80, 115, 22, 120}
{115, 22, 120}
{22, 120}
{120}
set()알고리즘
1단계: 세트를 생성합니다. 2단계: pop() 함수를 사용합니다. 이 메서드는 세트에서 임의의 요소를 제거하고 해당 요소를 반환합니다. 3단계: 세트가 빌 때까지 위 과정을 반복합니다.
예제 코드
# 세트에서 요소를 제거하는 파이썬 프로그램
def remove_element(set1):
print("NEW SET IS ::>")
while set1:
set1.pop()
print(set1)
# 드라이버 코드
set1 = set([22, 120, 130, 115, 80, 9])
remove_element(set1)실행 결과
NEW SET IS ::>
{9, 80, 115, 22, 120}
{80, 115, 22, 120}
{115, 22, 120}
{22, 120}
{120}
set()위 코드에서는 while set1: 조건을 사용해 세트가 비어 있지 않은 동안 계속 pop()을 호출합니다. 매번 요소가 하나씩 제거되며, 마지막에는 모든 요소가 사라져 빈 세트인 set()이 출력되는 것을 확인할 수 있습니다.
참고: 세트에서 요소를 제거하는 다른 방법들
pop() 외에도 세트의 요소를 제거하는 몇 가지 유용한 메서드가 있습니다.
- remove(x): 지정한 요소 x를 제거합니다. 단, 해당 요소가 세트에 존재하지 않으면 KeyError 오류가 발생합니다.
- discard(x): 지정한 요소 x를 제거하지만, 요소가 존재하지 않아도 오류를 발생시키지 않습니다.
- clear(): 세트의 모든 요소를 한 번에 제거하여 빈 세트로 만듭니다.
상황에 맞게 이러한 메서드들을 활용하면 세트 데이터를 더욱 효율적으로 관리할 수 있습니다.