[Python] 람다 (lambda)

개념

람다(lambda)란?

익명 함수

 Python에서는 람다 함수를 통해 이름 없는 함수를 만들 수 있다.

 

람다 함수의 장점

• 코드의 간결함
• 메모리의 절약

 

람다 함수는 정의와 동시에 사용할 수 있는 반면, 함수에 이름이 없고 저장된 변수가 없기 때문에 다시 사용할 수는 없다.

 

📍

cf) def 활용한 함수 생성 : 리터럴 표기법에 따른 함수 생성방법. 클래스를 통해 생성된 객체 인스턴스이므로 일회성 계산에는 메모리 낭비가 될 수 있다.

 

 

활용 방법

 

람다 함수는 lambda 매개변수 : 표현식 의 형식으로 사용하면 된다.

예시 1을 통해 위에서 말한 def 를 활용한 함수 생성 방법과 lambda를 활용한 함수 생성 방법을 비교해보자.

 

활용 예시 1. 인수로 받은 값에 10을 더해 반환하기

1)  def 함수

def plus_ten(x):
	return x + 10

print(plus_ten(1))		# => 11

2)  lambda 익명함수

print((lambda x: x + 10)(1))		# => 11

이렇게 람다함수는 람다 표현식 자체를 호출할 수도 있고,

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

이렇게 람다표현식을 변수에 할당해줄 수 도 있다. 이는 1)처럼 함수로 활용할 수 있는데, 그림으로 비교하면 보다 쉽게 이해할 수 있다.

---

💡  def 함수 vs lambda 익명함수

💡

주의! 람다 표현식 안에서는 새 변수를 만들 수 없다. 단, 람다 표현식 밖의 변수는 사용할 수 있다.

# 람다 표현식 내에 변수 지정 -> Error 
print(lambda x: y=10; x+y)(1)) 		# => SyntaxError: invalid syntax

# 람다 표현식 외부의 변수 사용
z = 10
print((lambda x: x + z)(1)) 		# => 11

---

 

활용 예시 2. 두 요소의 합을 구하기

print((lambda x,y: x + y)(10, 20))			# => 30

 

예시 3. 앞뒤 불필요한 공백을 제외한 문자의 길이로 정렬하기

target = ['  cat ', ' tiger ', '    dog', 'snake   ']
print(sorted(target, key=lambda x : len(x.strip())))

 

 

 

여러 함수와의 활용

 

람다 표현식을 사용하는 이유는 함수의 인수 부분에서 간단하게 함수를 만들기 위해서이다. 이런 방식으로 사용하는 대표적인 예가 map이다. 우선  map 함수에 대해 정확히 이해하기 위하여, def 함수를 활용하여 map의 활용 예시를 살펴보자.

1. map()

map(함수, 리스트)     →    리스트의 원소들을 하나씩 꺼내 함수를 적용시킨 후, 새로운 리스트에 담아준다.

1-1. 활용 :  lambda + map()

def plus_ten(x):
	return x + 10

print(list(map(plus_ten, [1, 2, 3])))		# => [11, 12, 13]

보통 map함수를 활용할 때에는 map(str, [1, 2, 3])과 같이 자료형을 변경할 때에만 자주 활용했는데, 이렇게 함수를 직접 만들어서 넣어도 무방하다.

1-2. 활용 :  lambda + map()

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

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

전체적으로 보면 1-1에 비하여 코드가 세 줄에서 한 줄로 줄어든 것을 확인할 수 있다. 이처럼 람다 표현식은 함수를 다른 함수의 인수로 넣을 때 매우 편리하다.

 

---

💡 여기서 잠깐!  함수의 인수와 인자란?

• 인자 (parameter) : 함수를 정의할 때 외부로부터 받아들이는 임의의 값. 위의 그림에서 인자값은 x, y (인자값 = 매개변수)
• 인수 (argument) : 함수를 호출할 때 사용되는 값. 위의 그림에서 인수는 2, 5

---

 

 

 

📍

람다 표현식으로 매개변수가 없는 함수도 만들 수 있다. 단, 콜론 뒤에는 반드시 반환할 값이 있어야 함!

# 매개 변수가 없는 람다 표현식

(lambda : 1)()			# => 1

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

 

 

2. reduce() 

 

❗

Python3에서는 from functools import reduce 필수!

    reduce라는 단어와 연산 방식이 매치가 잘 되지 않지만, 시퀀스의 원소가 누적되며 줄어들기 때문이라고 이해하면 쉽다.

reduce(함수, 시퀀스)       :  시퀀스(문자열, 리스트, 튜플)의 원소들을 누적적으로 함수에 적용시킨다.

from functools import reduce

reduce(lambda x, y: x + y, [0, 1, 2, 3, 4])			# => 10

 

3. filter() 

filter(함수, 리스트)              :  리스트의 원소들을 함수에 적용시켜서 결과가 참인 값들로 새로운 리스트를 리턴

# 5 이하의 원소만 필터링하기

list(filter(lambda x: x < 5, range(10))) 		# => [0, 1, 2, 3, 4]

# 홀수만 필터링하기

list(filter(lambda x: x % 2, range(10)))  	# => [1, 3, 5, 7, 9]

 

 

---

참고 사이트

• 제대로 파이썬 (https://wikidocs.net/22804)
• 왕초보를 위한 Python: 쉽게 풀어 쓴 기초 문법과 실습 (https://wikidocs.net/64)
• 코딩 도장 (https://dojang.io/mod/page/view.php?id=2359)