¿Pasar funciones con argumentos a otra función en Python?
Pregunta
¿Es posible pasar funciones con argumentos a otra función en Python?
Di algo como:
def perform(function):
return function()
Pero las funciones que se pasarán tendrán argumentos como:
action1()
action2(p)
action3(p,r)
Solución
¿Te refieres a esto?
def perform( fun, *args ):
fun( *args )
def action1( args ):
something
def action2( args ):
something
perform( action1 )
perform( action2, p )
perform( action3, p, r )
Otros consejos
Para esto es lambda:
def Perform(f):
f()
Perform(lambda: Action1())
Perform(lambda: Action2(p))
Perform(lambda: Action3(p, r))
Puede usar la función parcial de functools de esta manera.
from functools import partial
def perform(f):
f()
perform(Action1)
perform(partial(Action2, p))
perform(partial(Action3, p, r))
También funciona con palabras clave
perform(partial(Action4, param1=p))
¡Usa functools.partial, no lambdas! Y ofc Perform es una función inútil, puede pasar funciones directamente.
for func in [Action1, partial(Action2, p), partial(Action3, p, r)]:
func()
(meses después) un pequeño ejemplo real donde lambda es útil, parcial no:
supongamos que desea varias secciones transversales unidimensionales a través de una función bidimensional,
como rodajas en una hilera de colinas.
quadf (x, f)
toma un f
1-d y lo llama para varios x
.
Para llamarlo para cortes verticales en y = -1 0 1 y cortes horizontales en x = -1 0 1,
fx1 = quadf( x, lambda x: f( x, 1 ))
fx0 = quadf( x, lambda x: f( x, 0 ))
fx_1 = quadf( x, lambda x: f( x, -1 ))
fxy = parabola( y, fx_1, fx0, fx1 )
f_1y = quadf( y, lambda y: f( -1, y ))
f0y = quadf( y, lambda y: f( 0, y ))
f1y = quadf( y, lambda y: f( 1, y ))
fyx = parabola( x, f_1y, f0y, f1y )
Hasta donde yo sé, parcial
no puede hacer esto -
quadf( y, partial( f, x=1 ))
TypeError: f() got multiple values for keyword argument 'x'
(¿Cómo agregar etiquetas numpy, parcial, lambda a esto?)
Esto se llama funciones parciales y hay al menos 3 formas de hacerlo. Mi forma favorita es usar lambda porque evita la dependencia de un paquete adicional y es el menos detallado. Suponga que tiene una función add (x, y)
y desea pasar add (3, y)
a alguna otra función como parámetro, de modo que la otra función decida el valor para y
.
Usar lambda
# generic function takes op and its argument
def runOp(op, val):
return op(val)
# declare full function
def add(x, y):
return x+y
# run example
def main():
f = lambda y: add(3, y)
result = runOp(f, 1) # is 4
Crea tu propia envoltura
Aquí debe crear una función que devuelva la función parcial. Obviamente, esto es mucho más detallado.
# generic function takes op and its argument
def runOp(op, val):
return op(val)
# declare full function
def add(x, y):
return x+y
# declare partial function
def addPartial(x):
def _wrapper(y):
return add(x, y)
return _wrapper
# run example
def main():
f = addPartial(3)
result = runOp(f, 1) # is 4
Usar parcial de functools
Esto es casi idéntico a lambda
que se muestra arriba. Entonces, ¿por qué necesitamos esto? Hay algunas razones . En resumen, parcial
puede ser un poco más rápido en algunos casos (consulte su implementación ) y que puede usarlo para la unión temprana versus la unión tardía de lambda.
from functools import partial
# generic function takes op and its argument
def runOp(op, val):
return op(val)
# declare full function
def add(x, y):
return x+y
# run example
def main():
f = partial(add, 3)
result = runOp(f, 1) # is 4
Aquí hay una manera de hacerlo con un cierre:
def generate_add_mult_func(func):
def function_generator(x):
return reduce(func,range(1,x))
return function_generator
def add(x,y):
return x+y
def mult(x,y):
return x*y
adding=generate_add_mult_func(add)
multiplying=generate_add_mult_func(mult)
print adding(10)
print multiplying(10)