Heim > Fragen und Antworten > Hauptteil
[('d', 100), ('c', 99), ('a', 89), ('b', 86)]如何快速得出['d','c','a','b']
arr1 = [('d', 100), ('c', 99), ('a', 89), ('b', 86)]
arr2 = []
for key in range(len(arr)):
arr2.append(arr[key][0])
感觉好low,有什么好的办法吗?
其实在用Python中在刷leetcode来学习
题目:https://leetcode.com/problems/top-k-frequent-elements/
class Solution(object):
def topKFrequent(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: List[int]
"""
items = {}
for item in nums:
if items.has_key(item):
items[item]+=1
else:
items[item]=1
arr1 = sorted(items.iteritems(), key=lambda asd:asd[1], reverse=True)
arr2 = []
for key in range(len(arr1)):
arr2.append(arr1[key][0])
return arr2[0:k]
怪我咯2017-04-17 17:45:28
很簡單,使用 list comprehension 即可:
arr1 = [('d', 100), ('c', 99), ('a', 89), ('b', 86)]
arr2 = [pair[0] for pair in arr1]
看了一下你原本的問題,寫了一個簡潔的版本,可以參考一下:
(適用於 Python2.7+, Python3)
from collections import Counter
def top_k_frequent(lst, k):
return [key for key, count in Counter(lst).most_common(k)]
使用:
lst = [1, 1, 1, 2, 3, 4, 4]
print(top_k_frequent(lst, 2))
效果:
[1, 4]
說明:
Python2.7+之後的版本,在 collections
庫裡有一種類 Counter
可以用。
詳細的操作方法請參考Counter object
利用 Counter(lst)
可以輕鬆得到一個 Counter
實例,裡面已經對 lst
中的元素作過統計了。
之後利用 most_common(k)
方法可以輕鬆得到一個排序過的 list of tuple,而且只會剩下前出現頻率前k高的項目,最後用 list comprehension
取出元素本身:
$>> from collections import Counter
$>> counter = Counter([1, 1, 1, 2, 3, 4, 4])
$>> counter
Counter({1: 3, 4: 2, 2: 1, 3: 1})
$>> most_items = counter.most_common(2)
$>> most_items
[(1, 3), (4, 2)]
$>> [key for key, count in most_items]
[1, 4]