파이썬(python) - 기본 문법 정리 (3)

사용자 정의로 리스트 정렬하기

 

리스트를 정렬할때는 list.sort() 함수와 sorted() 함수를 사용합니다. 

 

둘의 차이점은 리스트 정렬하기 에서 소개하고 있습니다. 

 

리스트를 정렬할때 사용자 정의로 정렬할 수 있습니다.

 

예를들어 리스트가 다음과 같이 있을 때 

 

>>> li = [[2,3],[1,5],[4,1],[1,4],[3,2]]

 

sort()함수를 사용하면 다음과 같이 정렬됩니다.

 

>>> li.sort()
>>> li
[[1, 4], [1, 5], [2, 3], [3, 2], [4, 1]]

 

sort() 함수는 자동적으로 리스트의 각 아이템의 첫 번째 요소 순으로 정렬된 것을 알 수 있습니다.

 

여기서 두 번째 요소 순으로 정렬하고 싶다면 사용자 정의로 key 인자에 내가 커스텀할 함수를 비교 함수를 넘겨주면 됩니다.

 

>>> li.sort(key = lambda x : x[1])
>>> li
[[4, 1], [3, 2], [2, 3], [1, 4], [1, 5]]

 

---

집합 자료형 set()

 

set 은 집합에 관련된 것을 쉽게 처리하기 위한 자료형입니다.

 

집합 자료형은 {} 로 표현됩니다.

 

>>> se = set(a)
>>> se
{1, 2, 3}

 

 

집합과 관련된 합집합, 교집합, 차집합을 표현할 수 있습니다.

 

>>> a = [1,2,3]
>>> b = [3,4,5]
>>> set(a) | set(b)
{1, 2, 3, 4, 5}
>>> set(a) & set(b)
{3}
>>> set(a) - set(b)
{1, 2}

 

리스트를 자주 사용하기 때문에 결과값을 list() 함수를 이용해 리스트로 변환해서 사용하면 유용합니다.

 

>>> a = [1,2,3]
>>> b = [3,4,5]
>>> ab = list(set(a) & set(b))
>>> ab
[3]

 

---

제곱 구하기

 

>>> 2**3
8