[python 기초] Code-kata week 2-4

🖥 Code-kata week2-4

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

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

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

nums = [1], k = 1
return [1]

풀이

1 첫 번째 접근 방식

• 각 숫자의 개수를 센다: set(nums) > count 적용
• dictionary형태로 만들어서 오름차순 후, k만큼 return 한다

def top_k(nums, k):
  std_list = list(set(nums))
  my_dict  = {}
  cnt      = 0 
  #{i:cnt}형태의 key-value 값 생성 
  for i in std_list:
      cnt = nums.count(i)
      my_dict[i] = cnt
  
  #my_dict의 value를 기준으로 정렬 후, 오름차순 -> tuple로 출력됨 
  sorted_dict = sorted(my_dict.items(), key = lambda item:item[1], reverse = True)
  
  #range(k)만큼 value값들을 return 
  result = []
  for i in range(k):
      result.append(sorted_dict[i][0])
  
  return result

2 두 번째 접근 방식

• 주어진 list 인자의 요소들을 차례대로 제거, 최후 남은 k개의 숫자를 반환
• 이중 for문으로 각 숫자를 제거

def top_k(nums, k):
	#총 len(nums)번 inner for문을 실행한다 
    for i in range(len(nums)):
    	#set(nums) 개수가 k개가 될까지 숫자를 차례대로 제거 
        for num in list(set(nums)): 
            if len(set(nums)) > k:
                nums.remove(num)
            else:
                list(set(nums))
    return list(set(nums))

additional notes)
lambda, sorted를 활용한 dictionary 정렬

#tuple 순서대로 정렬 
sorted(my_dict.items(), key = lambda x:x[0])

#value값 기준으로 정렬
sorted(my_dict.items(), key = lambda x:x[1]) 

그리고 위 문제는 zip을 활용해서 풀 수도 있다.

keys = [1,2,3]
values = ["한국","미국","프랑스"]
my_dict = dict(zip(keys,values))
>>> {1:"한국", 2:"미국", 3:"프랑스"}