[Python3] 집합 자료형에 대해 알아보자(set())

알고리즘 문제를 풀다 보면 대부분 딕셔너리나 리스트 자료형을 많이 이용하는 것 같지만, 아주 간혹 집합 자료형이 필요한 순간이 보인다. 그럴 때마다 집합 자료형 메서드를 찾아보거나 하는데, 한 번 개념을 정리하고 넘어가고 싶어서 공부한 내용을 정리하게 되었다.

---

집합 자료형의 선언

set1 = set([1, 2, 3]) # {1, 2, 3}
set2 = set("Grace") # {'G', 'r', 'a', 'c', 'e'}
set3 = set()

set1 은 숫자형, set2는 문자형, 그리고 set3은 비어있는 집합 자료형을 표현해준 것이다.

 

집합 자료형의 특징

• 중복을 허용하지 않는다
• 순서가 없다(unordered)
• mutable(=값이 변하는) 객체이다

순서가 없기 때문에 중복을 제거하는 용도로 집합을 쓰기도 한다.

만약 집합을 인덱싱해서 접근하고 싶다면, 리스트 또는 튜플 형태로 바꾸어서 접근해야한다.

 

교집합, 합집합, 차집합

교집합

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

print(s1 & s2) # {4, 5, 6}
print(s1.intersection(s2)) # {4, 5, 6}

& 또는 intersection함수를 사용한다. s1.intersection(s2)나 s2.intersection(s1)이나 동일하다.

합집합

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

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

| 또는 union함수를 사용한다. s1.union(s2)나 s2.union(s1)이나 동일하다.

차집합

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

print(s1 - s2) # {1, 2, 3}
print(s2 - s1) # {8, 9, 7}
print(s1.difference(s2))
print(s2.difference(s1))

- 또는 difference함수를 사용한다. 차집합 특성상 s1.difference(s2)나 s2.difference(s1)는 다른 값이다.

그 외 함수

add함수

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

add(값)을 사용하면 값이 1개 추가된다.

update 함수

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

update를 하면 여러 개 값을 한 번에 넣을 수 있다.

remove 함수

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

remove(값)을 사용하면 특정 값을 삭제할 수 있다.

---

추가 함수나 메서드는 지속적으로 업데이트 할 예정입니다.