[Python] 사전에서 값을 기준으로 topK를 추출하는 방법

Python 사전(Dictionary)에서 값을 기준으로 topK를 추출하는 방법

python에서 dictionary를 갖고 있을때 (key, value)에서 value를 기반으로 topK를 뽑아내는 방법을 제시한다.
사전에서 값을 기준으로 정렬을 한 뒤에, [:N]을 이용해서 값을 추출할 수 있지만,
간단하게 한줄로 끝낼 수 있다. 기존 파이썬 패키지에 있는 heapq를 이용하면 된다.

heapq는 Heap queue algorithm으로 priority queue 알고리즘을 이용해서 값을 찾아낸다.
heaps은 binary tree를 이용하기 때문에 O(nlogn)

더 자세한 내용을 및 이론을 알고 싶으면

https://docs.python.org/2/library/heapq.html에서 8.4.3 Theory를 확인하면 좋다.

소스코드

	import heapq
	topK = 10
	item_score_dict = {'1': 3,'2':2,'3':5,'4':1, '8':3}
	ranklist = heapq.nlargest(topK, item_score_dict, key=item_score_dict.get)

view raw dict_topk_by_value.py hosted with ❤ by GitHub