🐹 컬렉션 타입 - 세트

오늘은 컬렉션 타입 중 하나인 세트(set)에 대해 알아보자.

💡 1. 세트

세트(set)는 중복되지 않은 항목들의 컬렉션이다. 중괄호({})와 set(리스트) 형태의 함수로 생성할 수 있다.

s1 = {1, 3, 5, 7}
print(s1) # 출력: {1, 3, 5, 7}
print(type(s1)) # 출력: <class 'set'>

s2 = {}
print(s2) # 출력:  {}
print(type(s2)) # 출력: <class 'dict'> 
# 데이터가 없으면 딕셔너리로 인식

s3 = set([1, 3, 5, 7])
print(s3) # 출력: {1, 3, 5, 7}
print(type(s3)) # 출력: <class 'set'>

li1 = [1, 2, 3, 4]
print(type(li1)) # 출력: <class 'list'>
s4 = set(li1)
print(type(s4)) # 출력: <class 'set'>

리스트의 중복 데이터를 제거하고 싶으면 set로의 변환을 통해 간단하게 중복 데이터들을 제거할 수 있다.

s5 = {1, 3, 5, 3, 7, 9, 1, 5, 10, 7}
print(s5) # 출력: {1, 3, 5, 7, 9, 10}

li2 = [1, 3, 5, 3, 7, 9, 1, 5, 10, 7]
s6 = set(li2)
print(s6) # 출력: {1, 3, 5, 7, 9, 10}

💡 2. 세트의 메서드

2-1. add()

add() 메서드는 세트의 요소를 추가한다.

s1 = {1, 3, 5, 7}
s1.add(2)
s1.add(4)
s1.add(3)
print(s1) # 출력: {1, 2, 3, 4, 5, 7}

2-2. update()

update() 메서드는 세트에 여러 개의 요소를 추가한다.

s1 = {1, 3, 5, 7}
s1.update([2, 4, 3])
print(s1) # 출력: {1, 2, 3, 4, 5, 7}

2-3. remove()

remove()는 세트의 요소를 제거한다. 단, 세트에 해당 요소가 없으면 에러가 발생한다.

s1 = {1, 3, 5, 7}
s1.remove(3)
print(s1) # 출력: {1, 5, 7}

s1.remove(4) # KeyError: 4

2-4. discard()

remove()와 마찬가지로 요소를 제거하지만, 요소가 없어도 에러가 발생하지 않는다.

s1 = {1, 3, 5, 7}
d1.discard(3)
print(s1) # 출력: {1, 5, 7}

s1.discard(4)
print(s1) # 출력: {1, 5, 7}

🐹: 그럼 그냥 discard()를 쓰면 되지 오류가 뜨는 remove()는 왜 사용하나요!!!!!
-> remove와 discard는 상황에 따라 구분되어 쓰인다. 요소가 없어도 상관없이 코드를 진행하고 싶다면 discard를 쓰면 되지만, 요소가 있는지 없는지 알아야하거나 오류 처리문을 사용할 땐 remove를 사용하면 된다.

2-5. copy()

세트를 복사하는 메서드이다. 두 세트 간 요소는 같지만 메모리 주소가 달라진다.

s1 = {1, 3, 5, 7}
s2 = s1.copy()

print(s1) # 출력: {1, 3, 5, 7}
print(s2) # 출력: {1, 3, 5, 7}

print(id(s1)) # 출력: 133746842319360
print(id(s2)) # 출력: 133746842320032

파이썬에서의 set는 집합에서 유래했다. 따라서 집합 계산이 가능하다. set의 집합 계산들에 대해 알아보자.

2-6. union()

합집합을 계산한다. 연산자 | 과 동일하다.

s1 = {10, 20, 30, 40, 50}
s2 = {30, 40, 50, 60, 70}

result1 = s1.union(s2)
print(result1) # 출력: {70, 40, 10, 50, 20, 60, 30}

result2 = s1 | s2 
print(result2) # 출력: {70, 40, 10, 50, 20, 60, 30}

2-7. intersection()

교집합을 계산한다. 연산자 & 와 동일하다.

s1 = {10, 20, 30, 40, 50}
s2 = {30, 40, 50, 60, 70}

result1 = s1.intersection(s2)
print(result1) # 출력: {40, 50, 30}

result2 = s1 & s2
print(result2) # 출력: {40, 50, 30}

2-8. difference()

차집합을 계산한다. 연산자 - 와 동일하다.

s1 = {10, 20, 30, 40, 50}
s2 = {30, 40, 50, 60, 70}

result1 = s1.difference(s2)
print(result1) # 출력: {10, 20}

result2 = s1 - s2
print(result2) # 출력: {10, 20}

2-9. symmetric_difference()

대칭 차집합을 계산한다. 연산자 ^ 와 동일하다.

s1 = {10, 20, 30, 40, 50}
s2 = {30, 40, 50, 60, 70}

result1 = s1.symmetric_difference(s2)
print(result1) # 출력: {20, 70, 10, 60}

result2 = s1 ^ s2
print(result2) # 출력: {20, 70, 10, 60}

2-10. set의 정렬

set는 순서가 없는 자료형이기 때문에 데이터를 순서대로 정렬하고 싶으면 리스트로 변환 후 sort, sorted를 사용하면 된다.

result = list(result2)
result.sort()
print(result) # 출력: [10, 20, 60, 70]