Python의 람다(lambda) 함수

오늘은 lambda 함수에 대해서 다뤄볼 예정이다

이 글도 마찬가지로 비전공자들도 이해하기 쉽게 작성해 볼 예정이다

 

람다(lambda) 함수란 무엇인가?

파이썬에서 함수를 만드는 방법은 아래 방법이 가장 익숙할 것이다

def add(x, y):
    return x + y

 

근데 이런 간단한 함수는 굳이 저렇게 만들지 않고 lambda로 만들 수 있다

add = lambda x, y: x + y

 

한마디로 lambda는 파이썬에서 함수를 간단하게 만드는 방법 중 하나이다.

 

람다(lambda)의 기본 형태

람다식의 기본 형태는 이렇게 생겼다

lambda 매개변수들: 표현식

add = lambda x, y: x + y

그래서 위에 코드처럼 x,  y가 매개변수들이고

표현식은 x + y이다

 

간단한 조건문을 사용하는 lambda

간단한 조건에 따라 값을 반환하는 경우에도 사용할 수 있다

예시:

check_positive = lambda x: "Positive" if x > 0 else "Non-positive"

print(check_positive(5))    # 출력: Positive
print(check_positive(-3))   # 출력: Non-positive
print(check_positive(0))    # 출력: Non-positive

 

 Key를 사용하는 람다식

일반적으로 sorted() 함수를 사용해서 리스트를 정렬할 때

정렬의 기준을 제공하기 위해서 람다식을 사용한다

 

예를 들어, 리스트 안에 있는 튜플들을 첫 번째 요소를 기준으로 정렬하고 싶다면?

data = [(3, 'apple'), (1, 'banana'), (2, 'orange')]

# 첫 번째 요소를 기준으로 정렬
sorted_data = sorted(data, key=lambda x: x[0])

print(sorted_data)

출력 값은 이렇게 나온다

[(1, 'banana'), (2, 'orange'), (3, 'apple')]

만약에 두 번째 요소를 기준으로 정렬하고 싶다면?

data = [(3, 'apple'), (1, 'banana'), (2, 'orange')]

# 두 번째 요소를 기준으로 정렬
sorted_data = sorted(data, key=lambda x: x[1])

print(sorted_data)

출력 값은?

[(3, 'apple'), (1, 'banana'), (2, 'orange')]