[파이썬/python] 집합

집합은 set을 사용해 만들 수 있다.

집합은 중복을 허용하지 않고 순서가 없다.

그렇기 때문에 인덱스가 없어 인덱싱을 통해 값을 얻기 위해서는 리스트나 튜플로 변환해야 한다.

s = set()
>>> {}

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

s = set("hello")
>>> {'e', 'h', 'l', 'o'}

집합에 값 1개 추가, 여러개 추가, 제거 등은 다음과 같이 구현할 수 있다.

s = set([1, 2, 3])

# 값 하나 추가
s.add(4)
>>> {1, 2, 3, 4}

# 값 여러개 추가
s.update([4, 5])
>>> {1, 2, 3, 4, 5}

# 특정 값 제거
s.remove(3)
>>> {1, 2}

 

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

기호를 이용하여 간단하게 구할 수 있다.

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

#교집합
s1 & s2
s1.intersection(s2)
>>> {3, 4, 5}

#합집합
s1 | s2
s1.union(s2)
>>> {1, 2, 3, 4, 5, 6, 7}

#차집합
s1 - s2
s1.difference(s2)
>>> {1, 2}