파이썬 람다 표현식(lambda expression)

람다 표현식

def로 함수를 정의해서 사용한 것과 달리 람다 표현식으로 익명 함수를 만들어 사용한다. 람다 표현식은 함수를 간편하게 작성할 수 있어서 다른 함수의 인수로 넣을 때 주로 사용한다.

람다 표현식으로 함수 만들기

lambda 매개변수들: 식

>>> def plus_ten(x):
...     return x + 10
...
>>> plus_ten(1)
11

# 위 함수를 람다 표현식으로 작성하면 이렇게 된다.
>>> lambda x: x + 10
<function <lambda> at 0x02C27270>

람다로 만든 익명 함수를 호출하려면 람다 표현식을 변수에 할당해주면 된다.

>>> plus_ten = lambda x: x + 10
>>> plus_ten(1)
11

변수에 할당하지 않고 람다 표현식 자체를 호출하기

(lambda 매개변수들: 식)(인수들)

>>> (lambda x: x + 10)(1)
11

람다 표현식으로 매개변수가 없는 함수 만들기

lambda 뒤에 아무것도 지정하지 않고 :(콜론)을 붙인다. 단, 콜론 뒤에는 반드시 반환할 값이 있어야 한다. 왜냐하면 표현식(expression)은 반드시 값으로 평가되어야 하기 때문이다.

>>> (lambda : 1)()
1
>>> x = 10
>>> (lambda : x)()
10

람다 표현식 안에서는 변수를 만들 수 없다

변수가 필요한 함수는 def로 작성하는 게 낫다.

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

람다 표현식 바깥에 있는 변수는 사용할 수 있다.

>>> y = 10
>>> (lambda x: x + y)(1)
11

람다 표현식을 인수로 사용하기

>>> list(map(lambda x: x + 10, [1, 2, 3]))
[11, 12, 13]

람다 표현식에 조건부 표현식 사용하기

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

>>> a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
>>> list(map(lambda x: str(x) if x % 3 == 0 else x, a))
[1, 2, '3', 4, 5, '6', 7, 8, '9', 10]

map은 리스트의 요소를 각각 처리하므로 lambda의 반환값도 요소여야한다.

람다 표현식 안에서 조건부 표현식 if, else를 사용할 때는 :(콜론)을 붙이지 않는다. 또한 elif를 사용할 수 없다. 조건부 표현식은 식1 if 조건식1 else 식2 if 조건식2 else 식3 형식처럼 if를 연속으로 사용해야 한다.

>> a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
>> list(map(lambda x: str(x) if x == 1 else float(x) if x == 2 else x + 10, a))
['1', 2.0, 13, 14, 15, 16, 17, 18, 19, 20]

이렇게 알아보기 힘든 코드는 def로 함수를 만들고 if, elif, else를 사용하는 것을 권장한다.

map에 객체를 여러 개 넣기

map은 리스트 등의 반복 가능한 객체를 여러 개 넣을 수도 있다.

>>> a = [1, 2, 3, 4, 5]
>>> b = [2, 4, 6, 8, 10]
>>> list(map(lambda x, y: x * y, a, b))
[2, 8, 18, 32, 50]

매개 변수와 넣어줄 반복 가능한 객체의 수를 맞춰준다.

filter 사용하기

filter는 반복 가능한 객체에서 특정 조건에 맞는 요소만 가져오는데, filter에 지정한 함수의 반환값이 True일 때만 해당 요소를 가져온다.
filter(함수, 반복가능한객체)

>>> def f(x):
...     return x > 5 and x < 10
...
>>> a = [8, 3, 2, 10, 15, 7, 1, 9, 0, 11]
>>> list(filter(f, a))
[8, 7, 9]

# 위 함수를 람다로 만든다.
>>> a = [8, 3, 2, 10, 15, 7, 1, 9, 0, 11]
>>> list(filter(lambda x: x > 5 and x < 10, a))
[8, 7, 9]

reduce 사용하기

reduce는 반복 가능한 객체의 각 요소를 지정된 함수로 처리한 뒤 이전 결과와 누적해서 반환하는 함수다.

from functools import reduce
reduce(함수, 반복가능한객체)

# 리스트에 저장된 요소를 순서대로 더한 뒤 누적된 결과를 반환
>>> def f(x, y):
...     return x + y
...
>>> a = [1, 2, 3, 4, 5]
>>> from functools import reduce
>>> reduce(f, a)
15

# 위 함수를 람다로 생성
>>> a = [1, 2, 3, 4, 5]
>>> from functools import reduce
>>> reduce(lambda x, y: x + y, a)
15

• map, filter, reduce와 리스트 표현식
  
  
  • 리스트(딕셔너리,세트) 표현식 > map, filter, 람다 표현식
    전자가 알아보기 쉽고 속도도 빠르다.
    
    
  • for, while문 > reduce
    reduce는 조금만 복잡해져도 가독성이 떨어진다.