使用Python中的字典调用带参数的函数

Question

我正在制作一个程序，它有一个主菜单，要求用户输入一个选项并将其存储在整数option1中，该整数在字典options中查找。然后运行相应的函数。如果函数没有参数，则以下代码有效：

options = {0 : FunctionZero,    # Assign functions to the dictionary
            1 : FunctionOne,
            2 : FunctionTwo,
            3 : FunctionThree}

options[option1]()    # Call the function

如果函数有参数，则上述代码不起作用，因为()部分假定函数没有参数，但我尝试了以下操作，将函数的名称和参数存储在字典中的元组中：

options = {0 : (FunctionZero,""),    # FunctionsZero, FunctionOne
            1 : (FunctionOne,""),    # and FunctionTwo have no parameters
            2 : (FunctionTwo,""),
            3 : (FunctionThree,True)}    # FunctionThree has one parameter

if options[option1][1] == "":    # Call the function
    options[option1][0]()
else:
    options[option1][0](options[option1][1])

这段代码似乎工作得很好，但我想知道是否有更好的方法来实现这一点，特别是如果函数需要几个参数？在其他语言中，比如C，我可能会使用switch或case语句（这不是Python中的语句），并且避免使用if...elif语句。

Answer 1

要添加到icktoofay's answer，如果要向lambda传递参数，请执行以下操作：

def printDouble( number ):
    print number * 2

options = {
    1: lambda num: printDouble(num)
}

options[1](4)    #this prints 8

通过在“：”之前添加lambda的参数，可以声明lambda接收到一个参数，然后在它调用的函数中使用该参数。

如果你不想使用lambdas，你可以用通常的方法

def printDouble( num ):
    print num * 2

def printHalf( num ):
    print half / 2

functionDictionary = {
    'Double': printDouble,
    'Half'  : printHalf
}

functionDictionary['Double'](2)    #This prints 4

Answer 2

当然。在Python中，函数可以采用位置参数或关键字参数。对于大多数函数，参数可以以任何一种方式传递，但这不一定适用于所有函数，因此我们确实需要将它们分开。位置参数在iterable（通常是列表或元组）中，关键字参数在从字符串到值的字典中。

然后，我们可以将每个函数表示为函数、位置参数和关键字参数的元组：

options = {
    0: (function_zero, [], {}),
    1: (function_one, [], {}),
    2: (function_two, [], {}),
    3: (function_three, [True], {}),
    4: (function_four, [], {'kwarg': True}),  # takes a keyword argument
}

你可以这样称呼他们：

func, args, kwargs = options[option1]
func(*args, **kwargs)

但是如果您总是要传入一个常量，那么有一个更好的方法：只需为每个调用函数的函数创建一个小的无参数包装器您希望如何调用它：

options = {
    0: function_zero,
    1: function_one,
    2: function_two,
    3: lambda: function_three(True),
    4: lambda: function_four(kwarg=True),
}

然后使用第一种方法：

options[option1]()

正如jornsharpe的答案中详细说明的，您还可以使用functools.partial，而不是lambda。正如他所指出的，这样做的好处是可以附加一些自己的论点：

options[option1]('hello')  # adds 'hello' to previously-specified arguments

不过，如果您不需要这个功能，零参数lambda将为您提供很好的服务。

Answer 3

在创建字典时，我将使用^{}来指定参数：

from functools import partial

options = {0: FunctionZero,   
           1: FunctionOne,    
           2: FunctionTwo,
           3: partial(FunctionThree, True)} 

注意，这还允许在调用函数时传递其他参数（只要字典中的所有函数在调用partial之后都缺少相同的参数）：

def test(one, two, three=None, four=None):
    ...

def test2(one, two, three=None):
    ...

options = {1: partial(test, 1, three=3, four=4),
           2: partial(test2, 1, three=3)}

...

options[choice](2) # pass the 'two' argument both functions still require