Code Kata(Python) week 2 - day 4

문제

nums는 숫자로 이루어진 배열입니다.

가장 자주 등장한 숫자를 k 개수만큼 return 해주세요.

nums = [1,1,1,2,2,3],
k = 2

return [1,2]

nums = [1]
k = 1

return [1]

문제 풀이

def top_k(nums, k):
    a = 0
    b = 0
    for i in nums:
        if nums.count(i) > a:
            a = nums.count(i)
            b = i
    return list(range(b, b+k))