[Python] 집합 자료형

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

집합 자료형 선언

set 키워드를 사용해 선언한다.

  • Empty 집합 자료형
    s = set()
    
  • 리스트를 파라미터로 넣을 수 있음
    >>> s1 = set([1,2,3])
    >>> s1
    {1, 2, 3}
    
  • 문자열을 파라미터로 넣을 수 있음
    >>> s2 = set("Hello")
    >>> s2
    {'e', 'H', 'l', 'o'}
    

집합 자료형의 특징

  • 중복을 허용하지 않는다.
  • 순서가 없다(Unordered).
리스트나 튜플은 순서가 있기 때문에 인덱싱을 통해 자료형의 키를 얻을 수 있다. But Set은 unordered 이기 때문에 인덱싱으로 값을 얻을 수 없다.

만약 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

교집합, 합집합, 차집합

  1. 교집합
    >>> s1 & s2
    >>> s1.intersection(s2)
    {4, 5, 6}
    
  2. 합집합
    >>> s1 | s2
    >>> s1.union(s2)
    {1, 2, 3, 4, 5, 6, 7, 8, 9}
    
  3. 차집합
    >>> s1 - s2
    {1, 2, 3}
    >>> s2 - s1
    {8, 9, 7}
    >>> s1.difference(s2)
    {1, 2, 3}
    >>> s2.difference(s1)
    {8, 9, 7}
    

Set 관련 함수들

  1. add 이미 만들어진 set 자료형에 값을 추가할 수 있다.
    >>> s1 = set([1, 2, 3])
    >>> s1.add(4)
    >>> s1
    {1, 2, 3, 4}
    

    여러개 추가하고자하면, update를 쓰면 된다.

    >>> s1 = set([1, 2, 3])
    >>> s1.update([4, 5, 6])
    >>> s1
    {1, 2, 3, 4, 5, 6}
    
  2. remove 특정 값을 제거하고 싶을 때는 다음과 같이 하면 된다.
    >>> s1 = set([1, 2, 3])
    >>> s1.remove(2)
    >>> s1
    {1, 3}
    

[참고자료] https://wikidocs.net/1015

댓글남기기