[python] 집합 자료형 set()

📗 집합 set()

set은 리스트와 비슷하게 볼 수 있다. 하지만 인덱스로 접근이 불가능하고, 정렬도 할 수 없다. for 문으로 하나하나 출력해봐도 그때 그때 순서가 뒤죽박죽으로 다르게 나온다.

 

set 을 사용하는 이유는

• 중복이 없다
• 특정 원소가 있는 지 확인할 때 O(1)의 시간 복잡도를 가진다. ( 리스트의 경우 O(N) ) ex)  if 원소 in 셋: 
• 집합 관련

🔵 교집합, 합집합, 차집합 

교집합: & or intersection

합집합: | or union

차집합: - or difference

set1 = {1, 2, 3, 4, 5}
set2 = {3, 4, 5, 6, 7}
# 교집합
print(set1.intersection(set2))  # {3, 4, 5}
print(set1 & set2)  # {3, 4, 5}
# 합집합
print(set1.union(set2))  # {1, 2, 3, 4, 5, 6, 7}
print(set1 | set2)  # {1, 2, 3, 4, 5, 6, 7}
# 차집합
print(set1.difference(set2))  # {1, 2}
print(set1 - set2)  # {1, 2}

 

🟢 선언 set

set을 선언할 때는 보통

my_set = set()
# or
my_set = {"apple", "melon", "grape"}
print(my_set)  # {'grape', 'melon', 'apple'}

🔺 위에서 출력한 것을 보면 순서가 다르게 나오는 것을 확인할 수 있다.

 

🟢 추가 (원소 한개 추가) add

my_set.add('orange')
print(my_set)  # {'orange', 'grape', 'melon', 'apple'}

 

🟢 추가 (원소 여러개 추가) update

my_set.update(["blueberry", "banana"])
print(my_set)  # {'orange', 'apple', 'banana', 'grape', 'melon', 'blueberry'}

 

🟢 삭제 remove

my_set.remove("apple")
print(my_set)  # {'blueberry', 'orange', 'grape', 'banana', 'melon'}

!!! 하지만 없는 원소를 지우려고 하면 KeyError 발생

my_set.remove("dog")

 

🟢 삭제2 discard

없는 원소를 지우려고 해도 에러가 나지 않는다.

my_set.discard("grape")
print(my_set)  # {'blueberry', 'melon', 'banana', 'orange'}
my_set.discard("dog")
print(my_set)  # {'blueberry', 'melon', 'banana', 'orange'}

🔺 remove는 원소가 꼭 있어야 하고, discard는 없어도 된다. (그럼 discard만 쓰면 되는거 아닐까? 😂😂 잘 모르겠다)

 

🟢 삭제와 뽑기 pop

pop은 사실 리스트의 가장 마지막에 있는 원소를 뽑아서 반환하는 함수이다. 하지만 set은 순서가 없기 때문에 어떤 값이 나올지 모른다. 똑같은 set에서 팝을 실행할 때마다 다른 값이 나온다.

# 1차 시도
print(my_set)  # {'banana', 'blueberry', 'orange', 'melon'}
tmp = my_set.pop()
print(tmp)  # banana
print(my_set)  # {'blueberry', 'orange', 'melon'}

# 2차 시도
print(my_set)  # {'blueberry', 'orange', 'banana', 'melon'}
tmp = my_set.pop()
print(tmp)  # blueberry
print(my_set)  # {'orange', 'banana', 'melon'}

🔺 자세히 보면 가장 앞의 원소가 나온 거 같다. 하지만!!! set은 순서가 없으므로 모든게 랜덤이다.

 

🟢 비우기 clear

set 안의 모든 원소를 지우는 메소드

my_set.clear()
print(my_set)  # set()

 

🟢 합치기 union

두개의 set을 합쳐서 새로운 set을 생성

your_set = {"tomato", "cherry", "lemon"}
print(my_set)  # {'melon', 'orange', 'blueberry'}
print(your_set)  # {'tomato', 'cherry', 'lemon'}
our_set = my_set.union(your_set)  
print(our_set)  # {'tomato', 'melon', 'orange', 'blueberry', 'cherry', 'lemon'}

 

🟢 합치기2 update

아까 위에서 봤던 여러개의 원소를 집어넣는 기능과 같다

여러개의 원소 대신 하나의 리스트나 셋을 집어 넣는다.

new_set = {"ddalgi"}
list1 = ['sagwa', 'podo']
new_set.update(list1)
print(new_set)  # {'podo', 'ddalgi', 'sagwa'}
set1 = {"bae", "subak"}
new_set.update(set1)
print(new_set)  # {'podo', 'ddalgi', 'sagwa', 'subak', 'bae'}