Python - Set 자료구조 소개 및 사용 방법

pythonbasic

Set는 순서가 보장되지 않는 Collection입니다.

Set에 대해서 자세히 알아보겠습니다.

1. Set 객체 생성(Create)

Set는 다음과 같이 {...}를 사용하여 생성합니다. {...}는 dict와 동일하지만 추가되는 아이템이 key 값을 갖고 있지 않습니다.

cities = {'Seoul', 'Tokyo', 'Beijing', 'Washington', 'Paris'}
print(cities)

출력해보면 순서가 보장되지 않는 것을 볼 수 있습니다.

{'Beijing', 'Seoul', 'Paris', 'Washington', 'Tokyo'}

주의할 점은 아이템이 없는 Set를 만들 때 다음과 같이 생성하면 안됩니다. { }는 dict인지 set인지 구분이 어렵습니다. { }는 기본적으로 dict 객체로 생성됩니다.

cities = {}
print(type(cities)) # <class 'dict'>

비어있는 Set 객체를 생성하고 싶다면 다음과 같이 set()으로 생성할 수 있습니다.

cities = set()
print(type(cities)) # <class 'set'>

2. Set의 데이터 접근(Access)

Set는 순서가 보장되지 않고 색인이 없는(unindexed) Collection이기 때문에 get()이나 [] 등으로 접근이 안됩니다.

다음과 같이 for in으로 Set의 모든 아이템에 접근할 수 있습니다.

cities = {'Seoul', 'Tokyo', 'Beijing', 'Washington', 'Paris'}
for city in cities:
    print(city)

Ouput:

Tokyo
Washington
Paris
Beijing
Seoul

또는 if in을 사용하여 특정 아이템이 Set에 있는지 확인할 수 있습니다.

cities = {'Seoul', 'Tokyo', 'Beijing', 'Washington', 'Paris'}

if 'Seoul' in cities:
    print('Seoul is in cities')

if 'Bangkok' not in cities:
    print('Bangkok is not in cities')

Output:

Seoul is in cities
Bangkok is not in cities

3. Set에 요소 추가(Add)

add()로 Set에 아이템을 추가할 수 있습니다.

cities = {'Seoul', 'Tokyo', 'Beijing', 'Washington', 'Paris'}
cities.add('Bangkok')
print(cities)

Output:

{'Washington', 'Tokyo', 'Paris', 'Beijing', 'Bangkok', 'Seoul'}

4. Set의 요소 업데이트(Update)

update()로 List의 모든 아이템을 Set에 추가할 수 있습니다.

cities = {'Seoul', 'Tokyo', 'Beijing'}
cities.update(['Bangkok', 'Washington', 'Paris'])
print(cities)

Output:

{'Bangkok', 'Paris', 'Tokyo', 'Washington', 'Seoul', 'Beijing'}

update()의 인자로 Set를 전달할 수도 있습니다. 중복을 허용하지 않기 때문에, 중복된 아이템이 있다면 하나만 Set에 등록이 됩니다.

cities = {'Seoul', 'Tokyo', 'Beijing'}
cities.update({'Seoul',  'Washington', 'Paris'})
print(cities)

Output:

{'Paris', 'Washington', 'Seoul', 'Tokyo', 'Beijing'}

다음과 같이 두개 이상의 Set를 업데이트할 수도 있습니다.

cities = {'Seoul', 'Tokyo', 'Beijing'}
cities.update({'Seoul',  'Washington'}, {'Paris', 'Bangkok'}, {'Hanoi'})
print(cities)

Output:

{'Beijing', 'Hanoi', 'Tokyo', 'Seoul', 'Paris', 'Washington', 'Bangkok'}

5. Set의 요소 삭제(Remove)

remove()로 Set에서 어떤 아이템을 삭제할 수 있습니다.

cities = {'Seoul', 'Tokyo', 'Beijing', 'Washington', 'Paris'}
cities.remove('Seoul')
print(cities)

Output:

{'Paris', 'Beijing', 'Washington', 'Tokyo'}

remove()는 존재하지 않는 아이템을 지우려고 하면 다음과 같이 에러가 발생합니다.

cities = {'Seoul', 'Tokyo', 'Beijing', 'Washington', 'Paris'}
cities.remove('Bangkok')
print(cities)

Output:

Traceback (most recent call last):
  File "/home/js/blog-code/python_example/set_example.py", line 40, in <module>
    cities.remove('Bangkok')
KeyError: 'Bangkok'

Discard

discard()도 remove()처럼 아이템을 Set에서 삭제할 수 있습니다. 차이점은 존재하지 않는 아이템을 삭제할 때 에러가 발생하지 않습니다.

cities = {'Seoul', 'Tokyo', 'Beijing', 'Washington', 'Paris'}
cities.discard('Seoul')
cities.discard('Bangkok')
print(cities)

Output:

{'Beijing', 'Washington', 'Tokyo', 'Paris'}

Pop

pop()은 Set의 마지막 아이템을 제거하고 그 값을 리턴합니다. 하지만 Set는 순서가 보장되지 않기 때문에 어떤 아이템이 삭제될지 예상할 수 없습니다.

cities = {'Seoul', 'Tokyo', 'Beijing', 'Washington', 'Paris'}
removed = cities.pop()
print(removed)
print(cities)

Output:

Washington
{'Beijing', 'Tokyo', 'Paris', 'Seoul'}

Del

del으로 Set 객체를 제거할 수 있습니다.

cities = {'Seoul', 'Tokyo', 'Beijing', 'Washington', 'Paris'}
del cities

6. Set의 합집합(Union)

union()은 합집합처럼, 두개의 Set를 하나의 Set로 합치고 그 Set를 리턴합니다. 중복을 허용하지 않기 때문에 중복된 값이 있다면 하나만 포함됩니다.

cities1 = {'Seoul', 'Tokyo', 'Beijing'}
cities2 = {'Seoul',  'Washington', 'Paris'}
cities3 = cities1.union(cities2)
print(cities3)

Output:

{'Seoul', 'Tokyo', 'Washington', 'Paris', 'Beijing'}

7. Set의 교집합(Intersection)

교집합처럼, 두 Set의 겹쳐지는 부분만 얻고 싶다면 다음과 같이 & 키워드를 사용할 수 있습니다.

cities1 = {'Seoul', 'Tokyo', 'Beijing'}
cities2 = {'Seoul',  'Washington', 'Paris'}
cities3 = cities1 & cities2
print(cities3)

Output:

{'Seoul'}
Loading script...

Related Posts

codechachaCopyright ©2019 codechacha