Python의 람다 함수

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

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

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

pythondef square(num):
  return num ** 2

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

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

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

python[1, 4, 9]

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

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

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

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

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

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

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

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

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

pythondef 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) )

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