[Python] 사전 자료형의 키, 값 기준으로 정렬하기 (sort a Dictionary by key, value)
Python 분석과 프로그래밍/Python 데이터 전처리 2019. 8. 28. 23:43이전 포스팅에서 사전 자료형(dictionary data type)을 만드는 방법과 기본 사용법(https://rfriend.tistory.com/333), 사전 자료형 내장함수와 메소드(https://rfriend.tistory.com/334)에 대해서 설명한 적이 있습니다.
사전 자료형을 중괄호로 묶여서 {키(key) : 값(value)} 의 쌍으로 이루어진다고 했으며, hash table type 으로 키를 해싱해놓고 있다가, 키를 기준으로 값을 찾으려고 하면 매우 빠른 속도로 찾아주는 효율적이고 편리한 자료형이라고 소개를 했었습니다.
이번 포스팅에서는 사전 자료형을
(1) 키를 기준으로 오름차순 정렬 (sort by key in ascending order)
(2) 키를 기준으로 내림차순 정렬 (sort by key in descending order)
(3) 값을 기준으로 오름차순 정렬 (sort by value in ascending order)
(4) 값을 기준으로 내림차순 정렬 (sort by value in descending order)
하는 방법을 알아보겠습니다.
{키 : 값} 쌍으로 이루어진 간단한 예제 사전 자료형을 만들어보겠습니다. (프로그래밍 언어별 인기도)
# make a dictionary pgm_lang = { "java": 20, "javascript": 8, "c": 7, "r": 4, "python": 28 } |
(1) 키를 기준으로 오름차순 정렬 (sort by key in ascending order): sorted() |
pgm_lang 사전형 자료를 키(key)를 기준으로 알파벳 오름차순으로 정렬해보겠습니다. dict.keys()를 정렬하면 keys 만 정렬된 값을 반환하며, dict.items()를 정렬하면 키(key)를 기준으로 정렬하되 키와 값을 튜플로 묶어서 정렬된 값을 반환합니다.
sorted(pgm_lang.keys()) ['c', 'java', 'javascript', 'python', 'r']sorted(pgm_lang.items()) [('c', 7), ('java', 20), ('javascript', 8), ('python', 28), ('r', 4)] |
그러면, 키를 기준으로 정렬한 (키, 값) 쌍의 튜플을 for loop 을 써서 한줄씩 인쇄를 해보겠습니다.
첫번째 예는 일반적인 for loop이며, 두번째는 list comprehension 을 이용하여 같은 결과로 인쇄한 예입니다.
for key, value in sorted(pgm_lang.items()): print(key, ":", value) c : 7
java : 20
javascript : 8
python : 28
r : 4# list comprehension [print(key, ":", value) for (key, value) in sorted(pgm_lang.items())] c : 7
java : 20
javascript : 8
python : 28
r : 4[None, None, None, None, None] |
키를 기준으로 정렬할 때, lambda 함수를 사용하여 새롭게 정의한 키(custom key function, logic)를 사용할 수도 있습니다. 아래 예는 키의 길이(length)를 기준으로 오름차순 정렬을 해본 것입니다.
# sort a dictionary by custom key function (eg. by the length of key strings) pgm_lang_len = sorted(pgm_lang.items(), key = lambda item: len(item[0])) # key: [0] for key, value in pgm_lang_len: print(key, ":", value) c : 7
r : 4
java : 20
python : 28
javascript : 8 |
(2) 키를 기준으로 내림차순 정렬 (sort by key in descending order): reverse=True |
내림차순으로 정렬하려면 reverse=True 옵션을 추가해주면 됩니다.
# sorting in reverse order sorted(pgm_lang.keys(), reverse=True) ['r', 'python', 'javascript', 'java', 'c']for (key, value) in sorted(pgm_lang.items(), reverse=True): print(key, ":", value) r : 4
python : 28
javascript : 8
java : 20
c : 7 |
(3) 값을 기준으로 오름차순 정렬 (sort by value in ascending order) |
값(value)을 기준으로 정렬하려면 앞서 소개했던 lambda 함수를 이용하여 키(key)로 사용할 기준이 값(value), 즉 item[1] 이라고 지정을 해주면 됩니다. (키는 item[0], 값은 item[1] 로 indexing)
sorted(pgm_lang.items(), key = lambda item: item[1]) # value: [1] [('r', 4), ('c', 7), ('javascript', 8), ('java', 20), ('python', 28)]
|
값을 기준으로 오름차순 정렬한 결과를 for loop을 같이 사용하여 (키 : 값) 한쌍씩 차례대로 프린트를 해보겠습니다. (list comprehension 을 사용해도 결과 동일)
pgm_lang_val = sorted(pgm_lang.items(), key = lambda item: item[1]) for key, value in pgm_lang_val: print(key, ":", value) r : 4
c : 7
javascript : 8
java : 20
python : 28# list comprehension [print(key, ":", value) for (key, value) in sorted(pgm_lang.items(), key = lambda item: item[1])] [None, None, None, None, None] |
(4) 값을 기준으로 내림차순 정렬 (sort by value in descending order) |
위의 (3)번과 같이 값을 기준으로 정렬하므로 key=lambda x:x[1] 로 값(value, 즉 x[1])이 정렬 기준이라고 지정을 해주구요, 대신 내림차순이므로 reverse=True 옵션을 추가해주면 됩니다.
# in reverse order pgm_lang_val_reverse = sorted(pgm_lang.items(), reverse=True, key=lambda item: item[1]) for key, value in pgm_lang_val_reverse: print(key, ":", value) python : 28
java : 20
javascript : 8
c : 7
r : 4 |
바로 위의 예에서는 for loop 순환할 때 key, value 2개 객체로 따로 따로 받았는데요, 바로 아래의 예처럼 items 로 (키, 값) 튜플로 받아서 items[0] 으로 키, items[1] 로 값을 indexing 할 수도 있습니다.
# the same result with above for items in pgm_lang_val_reverse: print(items[0], ":", items[1]) python : 28
java : 20
javascript : 8
c : 7
r : 4 |
이상으로 사전 자료형(dictionary)의 키, 값 기준 정렬(sorted)하는 4가지 방법 소개를 마치겠습니다.
- DataFrame, Tuple, List 정렬 방법은 https://rfriend.tistory.com/281 를 참고하세요.
- Numpy 배열 정렬 np.sort() 는 http://rfriend.tistory.com/357 를 참고하세요.
많은 도움이 되었기를 바랍니다.
이번 포스팅이 도움이 되었다면 아래의 '공감~
'를 꾹 눌러주세요. :-)
