[Python] lambda

아직도 람다식이 손에 안 익어가지고 알고리즘 20문제 풀어야 1번 쓸까 말까인데, 프로그래머스 효율성 문제 때문에 종종 람다식 써야 할 때가 있어서 이번 기회에 정리해보려고 한다.

lambda 표현식

lambda 매개변수: 식

조건부 표현식

lambda 매개변수: 식1 if 조건식1 else 조건식2

ex) list(map(lambda x: str(x) if x % 3 == 0 else x, a)) → a라는 리스트가 있다고 할 때 3의 배수인 값만 문자열로 변환

 

⭐ 주의 ⭐

- if를 사용했다면 else도 반드시 있어야 함

- elif를 사용할 수 없으며 if ~ else를 연속적으로 사용해야 함

def vs lambda

# def
def plus_ten(x):
    return x+10

# lambda expression
plus_ten = lambda x: x+10

∴ 같은 기능을 수행하지만 표현 방식이 다름

 

def 형태의 함수와 람다 표현식 비교(https://dojang.io/)

 

특징

- 람다식 안에서는 새로운 변수를 만들 수 없음

(lambda x: y = 10; x + y)(1)
SyntaxError: invalid syntax

 

- 식 밖의 변수는 사용 가능

y=10
(lambda x: x + y)(1)
# result = 11

 

- 매개변수가 없는 함수로 생성 가능 : (lambda: 반환값)( )

(lambda : 1)()
# result = 1

x = 10
(lambda : x)()
# result = 10

 

map/reduce/filter

1. map(람다식, 리스트)

list(map(lambdax: x**2, range(5)))
# result = [0,1,4,9,16]

 

2. reduce(함수, 문자열/튜플/리스트)

from functools import reduce
reduce(lambda x, y: x + y, [0, 1, 2, 3, 4])
# result = 10

 

* 10이 나오는 과정

첫번째 x, y : x=0, y=1 → x+y=1

두번째 x, y : x=1(첫번째 과정에서 x+y의 값), y=2 → x+y=3

세번째 x, y : x=3(두번째 과정에서 x+y의 값), y=3 → x+y=6

네번째 x, y : x=6(세번째 과정에서 x+y의 값), y=4 → x+y=10

 

 

3. filter(함수, 반복 가능한 객체) : 특정 조건에 맞는 요소를 이용해 새로운 객체 생성

list(filter(lambda x: x > 5 and x < 10, a))
# 반복 가능한 객체 a에서 5 초과 10 미만인 값만 가져옴

 

sort()/sorted()

- sort()와 sorted()가 가지고 있는 key 인자에 함수를 넘겨주면 정렬 기준은 함수의 반환값

- 정렬 기준이 여러 개일 경우는 튜플 형식으로 순서를 지정해주면 됨

 

ex) sorted(ex, key = lambda x : (x[1], -x[0]))에서의 정렬 순서

① ex의 두번째 요소를 이용해 오름차순 정렬

② ex의 첫번째 요소로 내림차순 정렬을 하겠다는 의미

 

 

 

🙏 틀린 내용이나 오타가 있다면 댓글로 편하게 알려주세요!