TIL31. Python : Lambda expression

📌 이 포스팅에서는 Python의 Lambda expression에 대해 정리하였습니다.

🌈 Lambda expression

🔥 lambda 표현식이란?

🔥 lambda 표현식의 활용

🔥 Assignment

1. lambda 표현식이란?

🤔 lambda 표현식은 함수의 축약형 표현식입니다.

✔️ lambda는 함수의 축약형 표현식으로 익명 함수 또는 람다 함수라 부릅니다.

✔️ 람다 표현식과 일반 함수의 차이점은 이름의 유무이며, return이 없어도 lambda 함수는 결과를 반환합니다.
✔️ lambda 함수는 다른 곳에서 재사용되지 않는 간단한 body를 갖춘 함수가 필요할 때 간편하게 사용됩니다.

print((lambda x,y : x // 2)(100,25)) # 50

🤔 lambda 표현식과 일반 함수의 비교

✔️ 아래 계산기 기능을 구현한 함수를 람다표현식으로 수정하면 더 간편하게 형태를 가진 것을 볼 수 있습니다.

# 일반함수
def add(x,y):
    return x+y
def sub(x,y):
    return x-y
def div(x,y):
    return x/y
def mul(x,y):
    return x*y
calculator = [add, sub, div, mul]
print(calculator[0](10,5)) # 15
print(calculator[1](10,5)) # 5
print(calculator[2](10,5)) # 2.0
print(calculator[3](10,5)) # 50

✔️ 아래처럼 간단한 로직을 lambda로 표현할 때, 더욱 직관적이기 때문에 가독성이 좋습니다.

# lambda 표현식
calculator = [
    lambda x,y : x+y,
    lambda x,y : x-y,
    lambda x,y : x/y,
    lambda x,y : x*y,
]
print(calculator[0](10,5)) # 15
print(calculator[1](10,5)) # 5
print(calculator[2](10,5)) # 2.0
print(calculator[3](10,5)) # 50

🤔 lambda 표현식은 표현식 내부에 변수를 선언할 수 없습니다.

✔️ lambda 표현식 내부에 변수를 선언하여 사용하면 SyntaxError가 발생합니다.

lambda_func = (lambda x: y=3; x//y)
print(lambda_func(27)) # SyntaxError: invalid syntax

✔️ 이에 어떤 변수를 lambda 표현식 내부로 전달하고 싶다면 parameter로 전달하던지 외부에 선언해주어 합니다.

# 변수 외부에 선언
y = 3
lambda_func = (lambda x: x//y)
print(lambda_func(27)) # 9

2. lambda 표현식의 활용

🤔 callback 함수와 lambda 표현식

✔️ iterable한 자료를 받아 요소를 제곱하는 함수는 아래와 같이 구현할 수 있습니다.

def func(x):
    return x**2
res = list(map(func, [1,2,3,4,5]))
print(res) # [1, 4, 9, 16, 25]

✔️ filter, map 등 콜백함수를 받아 처리할 때, lambda 표현식으로 구현하면 간편합니다.

res = list(map(lambda x: x**2, [1,2,3,4,5]))
print(res) # [1, 4, 9, 16, 25]

✔️ lambda 함수는 filter, map, zip 등과 함께 사용할 때 궁합이 좋습니다.

numbers = [1,2,3,4,5]
multiplier = [1,3,2,4,2]
squared = list(map(lambda x: x[0] ** x[1], zip(numbers, multiplier)))
print(squared) # [1, 8, 9, 256, 25]

🤔 lambda 표현식의 중첩

✔️ lambda 표현식은 중첩도 허용합니다. 단, 너무 깊은 중첩은 가독성이 떨어집니다.

numbers = [1,2,3,4,5]
multiplier = [1,3,2,4,2]
squared = list(filter(lambda x: x > 10, list(map(lambda x: x[0] ** x[1], zip(numbers, multiplier)))))
print(squared) # [256, 25]

3. Assignment

✔️ 다음은 비밀번호의 길이와 대문자가 포함된 것을 확인하는 간단한 함수입니다.

def check_password(password):
    if len(password) < 8:
        return 'SHORT_PASSWORD'
    if not any(c.isupper() for c in password):
        return 'NO_CAPITAL_LETTER_PASSWORD'
    return True
print(check_password('123')) # SHORT_PASSWORD
print(check_password('12356789f')) # NO_CAPITAL_LETTER_PASSWORD
print(check_password('123456789F')) # True

✔️ 위의 함수를 람다표현식으로 수정하여 구현하면 아래와 같습니다.
✔️ 람다표현식에서 if~else문도 축약([if일 때 body] if 조건 else [else일 때 body]) 가능합니다.

lambdas = [ 
    lambda x: 'SHORT_PASSWORD' if len(x) < 8 else None,
    lambda x: 'NO_CAPITAL_LETTER_PASSWORD' if not any(i.isupper() for i in x) else None   
]
def check_password_using_lambda(password):
    for f in lambdas:
        if f(password) is not None:
            return f(password)
    return True
print(check_password_using_lambda('123')) # SHORT_PASSWORD
print(check_password_using_lambda('12356789f')) # NO_CAPITAL_LETTER_PASSWORD
print(check_password_using_lambda('123456789F')) # True