Python의 람다 함수

리스트와 콜백을 매개변수로 받는 함수 func가 있다고 가정해 봅시다. 이 함수는 원본 리스트의 요소로부터 새로운 리스트를 생성할 것입니다. 이때 각 요소에는 콜백이 적용됩니다:

def func(lst, callback):
	res = []
	
	for el in lst:
		res.append(callback(el))
		
	return res

이제 전달된 숫자를 제곱하는 함수 square를 만들어 보겠습니다:

def square(num):
	return num ** 2

숫자 리스트와 콜백으로서의 함수 square를 func의 매개변수로 전달해 보겠습니다:

print( func([1, 2, 3], square) )

콘솔에는 숫자의 제곱으로 구성된 새로운 리스트가 출력됩니다:

[1, 4, 9]

하지만 이 코드는 람다 함수를 사용하여 상당히 줄일 수 있습니다. 람다 함수는 lambda 키워드로 호출되는 익명 함수로, 한 줄에 작성됩니다. 따라서 코드가 한 줄인 함수만 람다 함수로 다시 작성할 수 있습니다.

람다 함수의 구문은 다음과 같습니다:

lambda 함수의 매개변수: 매개변수와의 연산

람다 함수를 사용하여 함수 square를 다시 작성하고 func의 매개변수로 전달해 보겠습니다:

print( func([1, 2, 3], lambda num: num ** 2) )

람다 함수는 변수에 기록할 수 있습니다:

square = lambda num: num ** 2
print( func([1, 2, 3], square) )

다음 코드를 람다 함수를 사용하여 다시 작성하세요:

def func(num, clb):
	return clb(num)

def clb(num):
	return num + 1
	
print( func(2, clb) )

다음 코드를 람다 함수를 사용하여 다시 작성하세요:

def func(num, clb1, clb2):
	return (clb1(num), clb2(num))

def clb1(num):
	return num + 1
	
def clb2(num):
	return num - 1
	
print( func(2, clb1, clb2) )

다음 코드를 람다 함수를 사용하여 다시 작성하세요:

def func(num1, num2, clb):
	res = clb(num1) + num2
	return res
	
def clb(num):
	return num ** 3
	
print(func(2, 6, clb))