Função Lambda em Python
Suponha que temos uma função func,
que recebe como parâmetros uma lista e um
callback. A função criará uma nova
lista a partir dos elementos da lista original.
Além disso, o callback será aplicado
a cada um dos elementos:
def func(lst, callback):
res = []
for el in lst:
res.append(callback(el))
return res
Agora vamos criar uma função square,
que elevará ao quadrado o número passado para ela:
def square(num):
return num ** 2
Vamos passar para os parâmetros de func
uma lista numérica e a função square
como callback:
print( func([1, 2, 3], square) )
Um novo lista, consistindo dos quadrados dos números, será impressa no console:
[1, 4, 9]
No entanto, este código pode ser significativamente
encurtado usando uma função lambda
- uma função anônima, que é chamada com
a ajuda da palavra-chave lambda e
é escrita em uma única linha. Portanto, na
função lambda, só é possível reescrever
funções cujo código ocupa uma linha.
A sintaxe da função lambda é assim:
lambda parâmetro da função: operação com o parâmetro da função
Vamos reescrever a função square
usando uma função lambda e passá-la para o
parâmetro func:
print( func([1, 2, 3], lambda num: num ** 2) )
A função lambda pode ser atribuída a uma variável:
square = lambda num: num ** 2
print( func([1, 2, 3], square) )
Reescreva o seguinte código usando uma função lambda:
def func(num, clb):
return clb(num)
def clb(num):
return num + 1
print( func(2, clb) )
Reescreva o seguinte código usando uma função lambda:
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) )
Reescreva o seguinte código usando uma função lambda:
def func(num1, num2, clb):
res = clb(num1) + num2
return res
def clb(num):
return num ** 3
print(func(2, 6, clb))