1. 람다 표현식 사용하기
람다 표현식은 함수를 간편하게 작성할 수 있어서 다른 함수의 인수로 넣을 때 주로 사용합니다.
>>> def plus_ten(x):
... return x + 10
...
>>> plus_ten(1)
11
- lambda 매개변수들: 식
>>> lambda x: x + 10
<function <lambda> at 0x02C27270>
실행을 해보면 함수 객체가 나오는데, 이 상태로는 함수를 호출할 수 없습니다.
왜냐하면 람다 표현식은 이름이 없는 함수를 만들기 때문입니다.
그래서 람다 표현식을 익명 함수(anonymous function)로 부르기도 합니다.
lambda로 만든 익명 함수를 호출하려면 다음과 같이 람다 표현식을 변수에 할당해주면 됩니다.
>>> plus_ten = lambda x: x + 10
>>> plus_ten(1)
11
2. 람다 표현식 자체를 호출하기
람다 표현식은 변수에 할당하지 않고 람다 표현식 자체를 바로 호출할 수 있습니다. 다음과 같이 람다 표현식을 ( )(괄호)로 묶은 뒤에 다시 ( )를 붙이고 인수를 넣어서 호출하면 됩니다.
- (lambda 매개변수들: 식)(인수들)
>>> (lambda x: x + 10)(1)
11
람다 표현식 안에서는 변수를 만들 수 없다 단, 람다 표현식 바깥에 있는 변수는 사용할 수 있습니다.
>>> (lambda x: y = 10; x + y)(1)
SyntaxError: invalid syntax
>>> y = 10
>>> (lambda x: x + y)(1)
11
3. 람다 표현식을 인수로 사용하기
>>> def plus_ten(x):
... return x + 10
...
>>> list(map(plus_ten, [1, 2, 3]))
[11, 12, 13]
>>> list(map(lambda x: x + 10, [1, 2, 3]))
[11, 12, 13]
람다 표현식으로 매개변수가 없는 함수 만들기
>>> (lambda : 1)()
1
>>> x = 10
>>> (lambda : x)()
10
4. 람다 표현식과 map, filter, reduce 함수 활용하기
map -> map 객체 반환
filter -> filter 객체 반환
람다 표현식 작성 방법을 알아보았으니 이번에는 람다 표현식과 map, filter, reduce 함수를 함께 사용해보겠습니다.
4.1 람다 표현식에 조건부 표현식 사용하기
files = input().split()
print(list(map(lambda x : x.split('.')[0].zfill(3) + "." + x.split('.')[1] if len(x.split('.')[0])<3 else x,files)))
1.jpg 111.jpg 11.docx # 입력
['001.jpg', '111.jpg', '011.docx'] # 결과
먼저 람다 표현식에서 조건부 표현식을 사용하는 방법을 알아보겠습니다.
- lambda 매개변수들: 식1 if 조건식 else 식2
특히 람다 표현식에서 if를 사용했다면 반드시 else를 사용해야 합니다. 다음과 같이 if만 사용하면 문법 에러가 발생하므로 주의해야 합니다.
>>> 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]
>>> list(map(lambda x: str(x) if x % 3 == 0, a))
SyntaxError: invalid syntax
람다 표현식 안에서는 elif를 사용할 수 없습니다. 따라서 조건부 표현식은 식1 if 조건식1 else 식2 if 조건식2 else 식3 형식처럼 if를 연속으로 사용해야 합니다.
- lambda 매개변수들: 식1 if 조건식1 else 식2 if 조건식2 else 식3
>>> 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를 사용하는 것을 권장합니다.
>>> def f(x):
... if x == 1:
... return str(x)
... elif x == 2:
... return float(x)
... else:
... return x + 10
...
>>> a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
>>> list(map(f, a))
['1', 2.0, 13, 14, 15, 16, 17, 18, 19, 20]
4.2 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]
4.3 filter 사용하기
이번에는 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]
[연습문제]
files = [
"font",
"1.png",
"10.jpg",
"11.gif",
"2.jpg",
"3.png",
"table.xslx",
"spec.docx",
]
L = list(
filter(
lambda x: x.find(".jpg") != -1 or x.find(".png") != -1 or x.find(".gif") != -1,
files,
)
)
print(L) # ['1.png', '10.jpg', '11.gif', '2.jpg', '3.png']
4.4 reduce 사용하기
마지막으로 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
이제 함수 f를 람다 표현식으로 만들어서 reduce에 넣어보겠습니다.
>>> a = [1, 2, 3, 4, 5]
>>> from functools import reduce
>>> reduce(lambda x, y: x + y, a)
15
댓글