[python] lambda, sorted(iterable,key,reverse)

프로그래머스 알고리즘 고득점 KIT - 해시

 

1. 베스트 앨범

my_dict = {gen:[] for gen in set(genres)}
my_dict
for gen,play,idx in zip(genres, plays, range(len(plays))):
    my_dict[gen].append([play,idx])

genSorted = sorted(my_dict.keys(), key= lambda x: sum(map(lambda y: y[0],my_dict[x])), reverse=True)
genSorted
answer = []
for gen in genSorted:
    temp = [idx[1] for idx in sorted(my_dict[gen], key= lambda x:(x[0], -x[1]), reverse=True) ]
    answer += temp[:min(len(temp),2)]
answer

 

2. sorted란?

- sort()와 달리 기본 값을 유지하고, 정렬된 새로운 객체를 반환

- 리스트, 튜플, 문자열, 딕셔너리 등 정렬할 대상이 되는 iterable한 객체

- key: 정렬의 기준, 여러개가 있으면 정렬랄 함수를 넣는다 

- reverse: 내림차순

 

3. lambda란?

- def 키워드를 사용하지 않고 간단한 함수를 한 줄로 정의할 때 사용

- sort와 함께 쓸 경우, lambda x: 에서 x는 정렬할 객체의 요소를 가리킴

 예) ([('classic', 500), ('pop', 200)]) 를 정렬한다고 할때 ('classic', 500)와 ('pop', 200)

my_dict = {'classic': 500, 'pop': 200}

my_dict.items() # dict_items([('classic', 500), ('pop', 200)])

print(sorted(my_dict.items(), key=lambda x: x[1]))
# [('pop', 200), ('classic', 500)]

 

- 언제 사용할까?

1) 단순 계산

# 두 수를 더하는 함수
add = lambda x, y: x + y
print(add(3, 5))  # 출력: 8

 

2) 리스트 정렬

# 리스트를 두 번째 요소를 기준으로 정렬
data = [(1, 2), (3, 1), (5, 0)]
sorted_data = sorted(data, key=lambda x: x[1], reverse=True)
print(sorted_data)  # 출력: [(1, 2), (3, 1), (5, 0)]

 

3) map, filter와 함께 사용

# 리스트의 각 요소를 제곱
numbers = [1, 2, 3, 4]
squared = list(map(lambda x: x**2, numbers))
print(squared)  # 출력: [1, 4, 9, 16]

# 리스트에서 짝수만 필터링
evens = list(filter(lambda x: x % 2 == 0, numbers))
print(evens)  # 출력: [2, 4]

 

4) 함수 내부에서 사용(동적 함수 생성, 캡슐화 등)

def multiplier(n):
    return lambda x: x * n

times3 = multiplier(3)  # 3을 곱하는 함수 생성
times5 = multiplier(5)  # 5를 곱하는 함수 생성

print(times3(10))  # 출력: 30
print(times5(10))  # 출력: 50