将函数参数名称传递给Python中的另一个函数

Question

给出一个简单的函数：

def A(a = 1, b = 2):
    return a+b

我想编写另一个函数来更改a或b的默认参数值。 用户可以通过设置var = a或var = b来指定要更改的参数。 例如：

def B(var = 'a', new_value = 10):
    temp_dict = {var:new_value}
    ans = A(var)
    return ans    

要么

def B(var = 'a', new_value = 10):
    ans = A(var = new_value)
    return ans

在函数def B() ，设置var = a和var = new_value = 10 ，我期望A(var = new_value)可以达到与A(a = 10)相同的效果。 您知道编写函数def B()的正确方法吗？ 谢谢。

Answer 1

你快到了 通过B()函数，在调用A() ，需要解压缩temp_dict并将其作为参数传递给A() 。 见下文：

>>> def A(a = 1, b = 2):
...     return a+b
...

>>> def B(var = 'a', new_value = 10):
...     temp_dict = {var:new_value}
...     ans = A(**temp_dict)
        #        ^ unpack the dict, and pass it as an argument
...     return ans
...

>>> B()
12

---

有关此**如何与字典配合使用的更多详细信息，请查看：

• Python：* args和** kwargs？
• **（双星号/星号）和*（星号/星号）对参数有什么作用？

Answer 2

我可以随意解释OP想要的内容，即更改a或b的默认参数值 。 所以我要做的是返回转换后的函数A，其中a或b的默认值都通过部分更改：

from functools import partial

def B3(var ="a", new_value=10):
    return partial(A, **{var:new_value})

样本输出：

(Pdb) f = B3("a",10)
(Pdb) f()
12
(Pdb) f = B3("b",10)
(Pdb) f()
11
(Pdb) f(a=10)
20
(Pdb) f(b=13)
14
(Pdb) f(a=5,b=5)
10

但是，与请求的第二部分不同，该请求具有基于B(var="a",new_value=10)作为函数签名。

唯一的事情是，如果您不使用关键字参数，它会令人窒息：

(Pdb) f(7,7)
*** TypeError: A() got multiple values for keyword argument 'b'