top.py

class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        output = []
        counts = {}
        for i in nums:
            counts[i] = counts.get(i, 0) + 1
        for top in range(k):
            top_key = None
            top_value = 0
            for key, value in counts.items():
                if value > top_value:
                    top_value = value
                    top_key = key
            output.append(top_key)
            counts.pop(top_key)
        return output