如何编写返回另一个函数的函数?

时间:2013-01-10 15:34:44

标签: python function functional-programming currying

在Python中,我想编写一个返回另一个函数的函数make_cylinder_volume(r)。返回的函数应该可以使用参数h调用,并返回高度为h和半径为r的柱面的音量。

我知道如何从Python中的函数返回 values ,但是如何返回另一个函数

5 个答案:

答案 0 :(得分:142)

尝试使用Python:

import math
def make_cylinder_volume_func(r):
    def volume(h):
        return math.pi * r * r * h
    return volume

像这样使用它,例如使用radius=10height=5

volume_radius_10 = make_cylinder_volume_func(10)
volume_radius_10(5)
=> 1570.7963267948967

请注意,返回一个函数很简单,只需在函数内部定义一个新函数,并在最后返回它 - 小心地为每个函数传递适当的参数。仅供参考,从另一个函数返回函数的技术称为currying

答案 1 :(得分:12)

使用lambdas(也称为匿名函数),可以将volume内的make_cylinder_volume_func函数抽象为单行。与ÓscarLópez的答案完全不同,使用lambda的解决方案仍然在某种意义上“更具功能性”。

这是使用lambda表达式编写接受的答案的方法:

import math
def make_cylinder_volume_fun(r):
    return lambda h: math.pi * r * r * h

然后打电话给你任何其他咖喱功能:

volume_radius_1 = make_cylinder_volume_fun(1)
volume_radius_1(1) 
=> 3.141592653589793

答案 2 :(得分:10)

只想指出你可以用pymonad

来做到这一点
 import pymonad 

 @pymonad.curry
 def add(a, b):
     return a + b

 add5 = add(5)
 add5(4)
 9

答案 3 :(得分:1)

我知道我参加聚会已经太迟了,但是我认为您可能会觉得这种解决方案很有趣。

from math import pi
from functools import partial

def cylinder_volume(r, h):
    return pi * r * r * h

make_cylinder_with_radius_2 = partial(cylinder_volume, 2)
make_cylinder_with_height_3 = partial(cylinder_volume, h=3)

print(cylinder_volume(2, 3))            # 37.6991118431
print(make_cylinder_with_radius_2(3))   # 37.6991118431
print(make_cylinder_with_height_3(2))   # 37.6991118431

documentationpartial的工作方式。

答案 4 :(得分:-1)

这是一个庞大的示例,其中涵盖了一个函数中的许多单个多参数案例

def maths(var='NA'):
if var.lower() == 'add':
    def add(*args):
        return "Sum is : "+str(sum(args))
    return add
elif var.lower() == 'substract':
    def substract(a,b):
        if a>b:
            return "Difference is : "+str(a-b)
        else:
            return "Difference is : "+str(b-a)
    return substract
elif var.lower() == 'multiply':
    def multiply(*args):
        temp = 1
        for x in args:
            temp = temp*x
        return "multiplication is : "+str(temp)
    return multiply
elif var.lower() == 'divide':
    def divide(a,b):
        return "Division is : "+str(a/b)
    return divide
else:
    print("Please choose one of given operations: 'add','substract','multiply','divide'")

首先使用所需的操作调用maths函数,然后将返回的函数用于实际计算