347. Top K Frequent Elements

We can use Counter’s most common for this:

class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        return [n[0] for n in Counter(nums).most_common(k)]