[210729 TIL] Python

def top_k(nums, k):
    # 여기에 코드를 입력해주세요
  dict = {}
  for i in set(nums):
    dict[i] = nums.count(i)
  dict = sorted(dict.items(), key=lambda x: x[1], reverse=True)
  q = []
  for i in dict[:k]:
    q.append(i[0])
  return q

• nums에는 숫자 리스트가 들어오고 k에는 숫자가 들어온다.
  • nums에는 [1, 1, 1, 2, 2, 3] 이라는 리스트가 들어오고 k에는 2라는 숫자가 들어온다고 치자.
• 빈 딕셔너리 dict를 만든다.
• 숫자 리스트 nums를 set으로 만들어 중복되는 숫자를 제거한다.
  • set으로 리스트 nums의 중복되는 숫자를 제거하면 원소의 숫자만큼 dict안에 딕셔너리 키와 키값을 추가할 수 있다.
• 키는 중복되지 않는 숫자, 키 값은 리스트에서 숫자의 개수를 센 값을 넣는다

>>> dict
{1: 3, 2: 2, 3: 1}

• 이렇게 생성된 딕셔너리 dict에서 가장 큰 키값 두 개를 출력해야 하기 때문에 dict을 sort를 이용해 정렬해준다.
  • 딕셔너리를 정리할 때는 sorted를 사용하면 된다.

sorted

• sorted 함수의 경우 key 위치인자에 함수를 보내서, 함수에서 지정한 결과값에 따라서 정렬을 할 수 있다.