web-dev-qa-db-fra.com

Passer des fonctions avec des arguments à une autre fonction en Python?

Est-il possible de passer des fonctions avec des arguments à une autre fonction en Python?

Dis pour quelque chose comme:

def perform(function):
    return function()

Mais les fonctions à transmettre auront des arguments tels que:

action1()
action2(p)
action3(p,r)
pythonfunction
183
Joan Venge

Est-ce que tu veux dire ça?

def perform( fun, *args ):
    fun( *args )

def action1( args ):
    something

def action2( args ):
    something

perform( action1 )
perform( action2, p )
perform( action3, p, r )
268
S.Lott

C'est ce à quoi lambda est destiné:

def Perform(f):
    f()

Perform(lambda: Action1())
Perform(lambda: Action2(p))
Perform(lambda: Action3(p, r))
113
Dave

Vous pouvez utiliser la fonction partielle de functools comme ceci.

from functools import partial

def perform(f):
    f()

perform(Action1)
perform(partial(Action2, p))
perform(partial(Action3, p, r))

Fonctionne également avec des mots-clés

perform(partial(Action4, param1=p))
33
null

Utilisez functools.partial, pas lambdas! Et ofc Perform est une fonction inutile, vous pouvez transmettre des fonctions directement.

for func in [Action1, partial(Action2, p), partial(Action3, p, r)]:
  func()
13
Jochen Ritzel

(mois plus tard) un exemple réel minuscule où lambda est utile, partielle non:
Vous souhaitez différentes sections transversales à une dimension passant par une fonction à deux dimensions, comme des coupes à travers une rangée de collines.
quadf( x, f ) prend un 1-d f et l'appelle pour divers x.
Pour l'appeler pour les coupes verticales à y = -1 0 1 et les coupes horizontales à 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 )

Autant que je sache, partial ne peut pas faire cela -

quadf( y, partial( f, x=1 ))
TypeError: f() got multiple values for keyword argument 'x'

(Comment ajouter des tags numpy, partial, lambda à cela?)

6
denis

Cela s'appelle des fonctions partielles et il y a au moins 3 façons de le faire. Mon moyen préféré est d'utiliser lambda, car il évite la dépendance à un paquet supplémentaire et est le moins détaillé. Supposons que vous avez une fonction add(x, y) et que vous voulez transmettre add(3, y) à une autre fonction en tant que paramètre, de sorte que l'autre fonction décide de la valeur de y.

Utilisez 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

Créez votre propre wrapper

Ici, vous devez créer une fonction qui retourne la fonction partielle. C'est évidemment beaucoup plus verbeux.

# 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

Utilise partiel de functools

Ceci est presque identique à lambda ci-dessus. Alors pourquoi avons-nous besoin de cela? Il y a peu de raisons . En bref, partial pourrait être un peu plus rapide dans certains cas (voir son implementation ) et vous pouvez l'utiliser pour une liaison anticipée par rapport à la liaison tardive 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
4
Shital Shah

Voici un moyen de le faire avec une fermeture:

    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)
1
Stefan Gruenwald