[파이썬] 집합의 요소 추가 및 제거 연산

집합은 중복되지 않는 요소들의 모임입니다. 파이썬에서는 집합을 set 자료형으로 표현할 수 있으며, set 객체를 사용하여 요소를 추가하거나 제거할 수 있습니다. 이번 포스트에서는 파이썬에서 집합의 요소를 추가하고 제거하는 다양한 연산들을 살펴보겠습니다.

집합에 요소 추가하기

집합에 요소를 추가하기 위해서는 add 메서드를 사용합니다. 다음은 add 메서드를 사용하여 집합에 요소를 추가하는 예시입니다:

my_set = {1, 2, 3}
my_set.add(4)
print(my_set)

위 코드를 실행하면 다음과 같은 결과가 출력됩니다:

{1, 2, 3, 4}

add 메서드를 호출하여 4를 집합에 추가하였습니다.

집합에서 요소 제거하기

집합에서 요소를 제거하기 위해서는 remove 메서드를 사용합니다. 다음은 remove 메서드를 사용하여 집합에서 요소를 제거하는 예시입니다:

my_set = {1, 2, 3, 4}
my_set.remove(3)
print(my_set)

위 코드를 실행하면 다음과 같은 결과가 출력됩니다:

{1, 2, 4}

remove 메서드를 호출하여 집합에서 3이라는 요소를 제거하였습니다.

집합에서 존재하지 않는 요소 제거하기

remove 메서드를 사용할 때, 집합에 존재하지 않는 요소를 제거하려고 하면 KeyError가 발생합니다. 이를 방지하기 위해서는 discard 메서드를 사용할 수 있습니다. discard 메서드는 요소가 집합에 존재하지 않더라도 에러를 발생시키지 않는 점이 remove 메서드와 다릅니다.

다음은 discard 메서드를 사용하여 집합에서 요소를 제거하는 예시입니다:

my_set = {1, 2, 3, 4}
my_set.discard(5)
print(my_set)

위 코드를 실행하면 다음과 같은 결과가 출력됩니다:

{1, 2, 3, 4}

discard 메서드를 호출하여 집합에서 5라는 요소를 제거하려고 했지만, 집합에 5가 존재하지 않으므로 아무런 변화가 없습니다.

여러 개의 요소를 집합에 추가하기

여러 개의 요소를 집합에 한 번에 추가하려면 update 메서드를 사용할 수 있습니다. update 메서드는 iterable(반복 가능한)한 객체를 받아서 집합에 추가합니다. 다음은 update 메서드를 사용하여 집합에 여러 개의 요소를 추가하는 예시입니다:

my_set = {1, 2, 3}
my_set.update([4, 5, 6])
print(my_set)

위 코드를 실행하면 다음과 같은 결과가 출력됩니다:

{1, 2, 3, 4, 5, 6}

update 메서드를 호출하여 리스트 [4, 5, 6]의 요소들을 집합에 추가하였습니다.

집합의 요소 모두 제거하기

집합의 모든 요소를 제거하고 빈 집합으로 만들기 위해서는 clear 메서드를 사용합니다. 다음은 clear 메서드를 사용하여 집합을 비우는 예시입니다:

my_set = {1, 2, 3, 4, 5}
my_set.clear()
print(my_set)

위 코드를 실행하면 다음과 같은 결과가 출력됩니다:

set()

clear 메서드를 호출하여 집합의 모든 요소를 제거하였습니다.

위에서 설명한 연산들을 통해 파이썬에서 집합의 요소를 추가하고 제거하는 다양한 작업들을 할 수 있습니다. 집합은 중복되지 않는 요소들을 다루어야 할 때 유용한 자료형입니다. 파이썬은 집합을 쉽고 간편하게 다룰 수 있는 많은 메서드들을 제공합니다.