[Python] 집합 자료형

▶ 집합 자료형은 어떻게 만들까?

집합(set)은 파이썬 2.3부터 지원하기 시작한 자료형으로, 집합에 관련된 것을 쉽게 처리하기 위해 만든 자료형이다.

​

>>> s1 = set([1, 2, 3])
>>> s1
{1, 2, 3}

위와 같이 set()의 괄호 안에 리스트를 입력하여 만들 수 있다.

 

>>> s2 = set("Hello")
>>> s2
{'e', 'H', 'l', 'o'}

위와 같이 문자열을 입력해서 만들 수도 있다.

 

 

 

▶ 집합 자료형의 특징

set에는 2가지 큰 특징이 있다.

​

1. 중복을 허용하지 않는다.

2. 순서가 없다.

​

리스트나 튜플은 순서가 있기 때문에 인덱싱을 통해 자료형의 값을 얻을 수 있지만 set 자료형은 순서가 없기 때문에 인덱싱으로 값을 얻을 수 없다. 딕셔너리 역시 순서가 없는 자료형이라 인덱싱을 지원하지 않는다.

 

>>> s1 = set([1, 2, 3])
>>> l1 = list(s1)   ← 리스트로 변환
>>> l1
[1, 2, 3]
>>> l1[0]
1
>>> t1 = tuple(s1)   ← 튜플로 변환
>>> t1
(1, 2, 3)
>>> t1[0]
1

만약 set 자료형에 저장된 값을 인덱싱으로 접근하려면 다음과 같이 리스트나 튜플로 변환한 후 해야 한다.

​

 

 

▶ 교집합, 합집합, 차집합 구하기

set 자료형을 정말 유용하게 사용하는 경우는 교집합, 합집합, 차집합을 구할 때이다.

>>> s1 = set([1, 2, 3, 4, 5, 6])
>>> s2 = set([4, 5, 6, 7, 8, 9])

 

 

-교집합

>>> s1 & s2
{4, 5, 6}

'&' 기호를 사용하면 교집합을 간단히 구할 수 있다.

 

>>> s1.intersection(s2)
{4, 5, 6}

intersection 함수를 사용해도 동일한 결과를 돌려준다.

 

 

-합집합

>>> s1 | s2
{1, 2, 3, 4, 5, 6, 7, 8, 9}

'|' 기호를 사용한 방법

 

>>> s1.union(s2)
{1, 2, 3, 4, 5, 6, 7, 8, 9}

union 함수를 사용해도 동일한 결과를 돌려준다.

 

 

-차집합

>>> s1 - s2
{1, 2, 3}
>>> s2 - s1
{8, 9, 7}

빼기(-) 기호를 사용한 방법이다.

 

>>> s1.difference(s2)
{1, 2, 3}
>>> s2.difference(s1)
{8, 9, 7}

difference 함수를 사용해도 차집합을 구할 수 있다.

 

 

 

▶ 집합 자료형 관련 함수

​

-값 1개 추가하기(add)

이미 만들어진 set 자료형에 값을 추가할 수 있다.

>>> s1 = set([1, 2, 3])
>>> s1.add(4)
>>> s1
{1, 2, 3, 4}

1개의 값만 추가(add)할 경우에는 위와 같이 한다.

 

 

-값 여러 개 추가하기(update)

>>> s1 = set([1, 2, 3])
>>> s1.update([4, 5, 6])
>>> s1
{1, 2, 3, 4, 5, 6}

 

 

-특정 값 제거하기(remove)

>>> s1 = set([1, 2, 3])
>>> s1.remove(2)
>>> s1
{1, 3}