class Solution(object):
def topKFrequent(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: List[int]
"""
dict_num={}
for each in set(nums):
if each not in dict_num:
dict_num[each]=nums.count(each)
sort_list=sorted(dict_num.items(),key=lambda item:item[1],reverse=True)
out_list=[]
for i in range(k):
out_list.append(sort_list[i][0])
return out_list
修改后:
class Solution(object):
def topKFrequent(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: List[int]
"""
return zip(*collections.Counter(nums).most_common(k))[0]