Question

我觉得以前一定要问过这个问题，但我找不到，对不起，如果是这样的话！

我有一个不同参数的函数，如：

def f(arg1, arg2, ...,argN, kw1=0, kw2=0, kw3=0):
    # some code ...

假设我定义了我的（强制性）参数：

arg1 = some_value_1
arg2 = some_value_2
...
argN = some_value_N

我想缩小一些重复的函数调用，比如

f(arg1, arg2, ... argN, kw1=1)
f(arg1, arg2, ... argN, kw1=23)
f(arg1, arg2, ... argN, kw2=456)
f(arg1, arg2, ... argN, kw2=789)
f(arg1, arg2, ... argN, kw3='a')
f(arg1, arg2, ... argN, kw3='b')
...

看起来像：

for kw_def in [kw1=1, kw1=23, kw2=456, kw2=789, kw3='a', kw3='b',...]:
    f(arg1, arg2, ... argN, kw_def)

但语法正确。什么是正确的pythonic方式？

Answer 1

使用词典：

for kw_def in [
    {'kw1': 1},
    {'kw1': 23},
    {'kw2': 456},
]:
    f(arg1, ..., argN, **kw_def)

Answer 2

使用partial functions。

假设您有一个带有此签名的函数：

def f(arg1, arg2, arg3, kw1, kw2, kw3):
    pass

您可以像这样缩短具有arg1，arg2，kw1和kw2共同值的来电：

from functools import partial
f_short = partial(f, 1, 2, kw1="A", kw2="B")
f_short(3, kw3="C")

此f_short的来电与以下直接电话相同：

f(1, 2, 3, kw1="A", kw2="B", kw3="C")

如何迭代不同的kwargs？

2 个答案: