문제

nums는 숫자로 이루어진 배열입니다.
가장 자주 등장한 숫자를 k 개수만큼 return 해주세요.

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

그저께 풀었던 #7 리스트 내 과반수 넘는 숫자 반환하기 문제랑 비슷하다.
그 문제는 매개변수로 숫자리스트를 받아 과반수가 넘는 숫자를 바로 반환하면 되었지만,
이번 문제는 매개변수로 숫자리스트, k라는 매개변수 두개를 받아서 k의 갯수만큼의 숫자를 반환해야 했다.

#7번 문제랑 비슷하게 풀고 대신 k의 개수만큼을 반환하도록 하려 했다!

내가 푼 코드

def top_k(nums, k):
  count       = set(nums)
  check_nums  = {}
  final_nums  = []

  for i in count:
    check_nums[i] = nums.count(i)
  
  for j in range(k):
    max_key = max(check_nums, key=check_nums.get)
    final_nums.append(max_key)
    del(check_nums[max_key])

  return final_nums

우선 매개변수로 들어온 리스트를 set 자료구조형으로 바꿔서 유일한 숫자들을 추출한 뒤
해당 숫자만큼 for loop을 돌리면서 check_nums라는 dictionary 내부에
각 숫자, 그 숫자의 개수를 key, value구조로 넣어준다.

매개변수로 받은 k 만큼의 숫자들을 리턴해야 하기때문에

• 현재 check_nums라는 dictionary 내에 존재하는 value들 중 가장 큰 key값을 구한뒤,
  max_key값을 final_nums라는 list내에 넣고 check_nums라는 dictionary내에서 해당 key를 삭제한다.

는 조건으로 k만큼 반복하는 for loop을 돌린다.
이후 리턴해주기!

이것보다 효율적인 방법이 분명히 있을텐데....
다 풀어낸 뒤 모델 솔루션을 보기로 했다.

---

모델 솔루션

def top_k(nums, k):
    count = {}
    for n in nums:
        count[n] = count.get(n, 0) + 1
    bucket = [[] for _ in range(len(nums)+1)]		<- 여기!!
    for n, freq in count.items():
        bucket[freq].append(n)
    ret = []
    for n_list in bucket[::-1]:
        if n_list:
            ret.extend(n_list)
            if len(ret) == k:
                return ret

역시나 오늘의 모델 솔루션도 정말이지 쉽지 않다.. 하하
그래도 더욱 간결한, 수준높은 코드에 대한 이해도를 올리고 싶으니까..! 분석해보기🤯 🤯

but 분석하려고 했더니 당장 5번째 코드부터 막히는 것이 아닌가ㅠㅠ
당장 검색을 뭐라고 해야할 지.. 검색 키워드도 모르겠고..!(이럴때는 질문.. )

list comprehension이라는 선언 방식이라고 한다.
Advanced Python은 영영 안들어가볼줄 알았더니, 이렇게 보게 되네..
나 Python이랑 인연이었네..

암튼 그래서 list Comprehension이 무엇인가?

---

List Comprehension

• 새로운 리스트 생성할 때 사용할 수 있는 간단한 표현식
• 리스트와 같이 [ ] 대괄호를 사용하여 작성

• 우리가 만들려는 원소를 표현하는 표현식으로 시작하여 for loop이 뒤따라오는 형식
• for loop 뒤에 if문을 추가하여 조건문을 포함한 형식도 가능!

[ 표현식 for 원소 in 반복 가능한 객체 ]
[ 표현식 for 원소 in 반복 가능한 객체 if문 ]

👩🏻‍🦰 (과거의 하람) : 굳이 list comprehension이라는 개념을 알아야해? 그냥 for loop돌려서 넣으면 안돼?
👩‍💻 (현재의 나) : 응 안돼 훨씬 간결하게 코드를 짤 수 있으니까 꼭 알아둬야해

1) for list comprehension

1~10까지의 리스트 생성하고 싶을 때, 지금까지는 빈 배열을 선언하고, for loop을 선언해서, 빈 배열 안에 요소들이 추가되도록 하고.. 그걸 10번 반복을 하고..
아무리 간단하게 생각해봐도 3?4줄 정도의 코드가 나오는데 list comprehension을 사용하면?

new_list = [ x for x in range(1, 11) ]
print(new_list)

요로케 한 줄로 끝낼 수 있다. 🤩

2) if문을 포함하는 list comprehension

list내 숫자들 중 홀수만 추출해서 for loop 바깥의 odd_numbers라는 리스트에 append하는 코드의 경우,

odd_numbers = [ ]
for element in range(1,11):
    if (element % 2) == 1:
        odd_numbers.append(element)

지금까지는 이런 식으로 작성했었다!

if를 포함하는 list comprehension으로 수정해주면?

list_comprehension = [ element for element in range(1,11) if (element % 2) == 1 ]
print(list_comprehension)

한줄로 작성할 수 있게, 훨씬 간결해진다!

---

list comprehension의 장점은?

• 코드가 간결하다 & 작업 수행 속도가 훨씬 빠르다
• 데이터베이스를 조회하여 리스트로 만들 때 많이 사용된다
  : 새로운 배열을 만들때 유용
  👉🏻 DB에서 게시물 리스트를 가져올 때 속도를 훨씬 빠르게 진행할 수 있다!
  
  
• 또한 성능개선을 하기 위해서 리스트 컴프리헨션을 사용하기도 하지만 모든 상황에서 리스트 컴프리헨션이 만능은 아닙니다.
  예를 들어 코드의 가독성을 위해서는 여러 줄의 표현식과 조건문으로 표현하는 것이 이중 for문의 복잡한 리스트 컴프리헨션 한줄로 코딩하는 것보다 나은 경우도 있습니다.

👉🏻 구글의 파이썬 스타일 가이드에서 list comprehension의 장점과 단점을 확인 할 수 있다!