[Python] Set(셋)

 

파이썬에서 집합을 구현하고 싶을 땐 ?!

Set 을 이용하면 됩니다. 파이썬에서는 set이라는 집합 자료구조를 제공합니다. 

집합은 set( ) 과 같이 초기화 할 수 있습니다.

 

 

set 언제 사용하면 좋을까!?

집합은 이럴 때 사용하면 좋습니다.

1. 데이터의 중복이 없어도 될 때

집합에 담긴 데이터들은 모두 유니크 합니다. 즉, 집합에 5를 2번 넣어도 집합은 자신이 가진 원소 5는 1개라고 생각합니다.

 

2. 다루는 데이터의 삽입/삭제/검사가 자주 일어 날 때 특히 다루는 데이터가 정수가 아닐 때 

숫자형 데이터는 list를 사용하여 index를 통해 접근할 수 있습니다. 하지만 문자열,등 리스트에서 index로 활용할 수 없는 데이터를 빠르게 탐색시에는 set을 이용하면 좋습니다.

예를 들어, 리스트에 'aaa'원소가 있는지 알아내려면 O(N)이 걸리지만 집합은 O(1)이 소요됩니다.

 

3. 수학적으로 교집합, 차집합, 합집합을 구해야할 때 

파이썬의 집합에서는 수학에서의 집합이 가진 대부분의 기능을 제공합니다.

 

set의 시간복잡도

Operation	Time Complexity
탐색 "aaa" in myset	O(1)
제거 myset.remove	O(1)
합집합 set1 | set2	O(len(set1)+ len(set2))
교집합 set1 & set2	O(min(len(set1), len(set2))
차집합 set1 - set2	O(len(set1))
대칭차 set1 ^ set2	O(len(set1))

대칭차 = 합집합 - 교집합

 

set 사용시에 주의할 점

hashable 타입만 집합의 원소로 넣을 수 있습니다.

- hashable 타입 : int, float, decimal, bool, string, tuple, frozenset, 등

- non - hashable 타입 : list, dict,set, 유저가 만든 클래스, 등

 

set 사용법

📌 Init

set 함수를 호출하면 빈 집합을 선언할 수 있습니다.

# 빈 집합 생성하기

empty_set = set()
empty_set # set()

# 특정 원소를 가지는 집합 선언하기

my_set = set([1, 2, 3, 1, 2, 3])
my_set # {1,2,3}

 

📌 Add

집합에 원소를 넣을 때는 add 메소드를 이용합니다.

단 hashable 타입(int, float, string, 등 , list XX)만 집합에 넣을 수 있다는 것을 주의해야합니다!

# 값 집어넣기

my_set = set()
my_set.add(3)
my_set.add('동동')

my_set # {3,'동동'}

 

📌 Delete

집합에서 특정 원소를 지우려면 remove 메소드를 이용합니다.

# 원소 제거 예시
my_set = set([1,2,3,4,5])
my_set.remove(1)

my_set # {2,3,4,5}

 

📌 합집합 |

# 합집합: | 를 사용
set1 = set([1,2,3,4,5])
set2 = set([4,5,6,7,8])

set1 | set2 # {1, 2, 3, 4, 5, 6, 7, 8}

 

📌 차집합 -

# 차집합: - 를 사용

set1 = set([1,2,3,4,5])
set2 = set([4,5,6,7,8])

set1 - set2 # {1,2,3}

 

📌 교집합 &

# 교집합: & 를 사용

set1 = set([1,2,3,4,5])
set2 = set([4,5,6,7,8])

set1 & set2 #{4,5}

 

 

 

참고

TimeComplexity - Python Wiki