3.1 KiB
title |
---|
Lambda Expressions |
Lambda Expressions
Lambda Expressions are used when an operation only has to be performed once, meaning that there is no need for defining a function as it will not be used again. Lambda expressions also known as anonymous functions, as they are not named (defined).
Lambda functions can contain only one expression, so they are not well suited for functions with control-flow statements.
Syntax of Lambda Function
lambda arguments: expression
Lambda functions can have any number of arguments but only one expression.
Example code
# Lambda function to calculate square of a number
square = lambda x: x ** 2
print(square(3)) # Output: 9
# Traditional function to calculate square of a number
def square1(num):
return num ** 2
print(square1(5)) # Output: 25
In the above lambda example lambda x: x ** 2
yields an anonymous function object which can be associated with any name.
So, we associated the function object with square
and hence from now on we can call the object with square
like any traditional function. e.g. square(10)
Examples
Beginner
lambda_func = lambda x: x**2 # Function that takes an integer and returns its square
lambda_func(3) # Returns 9
Intermediate
lambda_func = lambda x: True if x**2 >= 10 else False # Function that returns True if the square of x >= 10
lambda_func(3) # Returns False (3**2 = 9, which is < 10)
lambda_func(4) # Returns True (4**2 = 16, which is >= 10)
Complex
my_dict = {"A": 1, "B": 2, "C": 3}
sorted(my_dict, key=lambda x: my_dict[x]%3) # Returns ['C', 'A', 'B'] # sort dict by the values % 3 (remainders from division by 3)
Passing lambda as fuction parameter
def apply(x, y, fun):
return fun(x, y)
res = apply(3, 5, lambda x, y: x + y)
print(res) # Output: 8
Use-case
Say you want to filter out odd numbers from a list
. You could use a for
loop:
my_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
filtered = []
for num in my_list:
if num % 2 != 0:
filtered.append(num)
print(filtered) # Python 2: print filtered
# [1, 3, 5, 7, 9]
You could write this as a one-liner with list-comprehensions
filtered = [x for x in [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] if x % 2 != 0]
However, another option is to use the built-in filter
function. Why? The first example is a bit to verbose, while the one-liner can be harder to understand. filter
offers the best of both words, and the built-in functions are usually faster.
my_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
filtered = filter(lambda x: x % 2 != 0, my_list)
list(filtered)
# [1, 3, 5, 7, 9]
NOTE: In Python 3 built in functions return generator objects, so you have to call list
, while in Python 2 they return a list
, tuple
or string
.
What happened? You told filter
to take each element in my_list
and apply the lambda expressions. The values that return False
are filtered out.