python - 更改python中函数的默认参数

Question

我有一个函数可以根据条件将 numpy 数组转换为包含 True 或 False 的数组，然后将相邻的 True 或 False 条目分组并计算每组的长度。这是为了确定给定月份降水数据中干期或湿期的长度。

这是函数:

import itertools
def spell(X, kind='wet', how='mean', threshold=0.5): 

    if kind=='wet':
        condition = X>threshold
    else:
        condition = X<=threshold

    length = [sum(1 if x==True else nan for x in group) for key,group in itertools.groupby(condition) if key]

    if not length: 
        res = 0
    elif how=='mean': 
        res = np.mean(length)
    else:
        res = np.max(length)

    return res

因此，基本上可以选择在给定降水数据的 numpy 数组的情况下确定湿期或旱期的平均长度或最大长度，默认参数设置为湿期平均长度。

我将这个函数与 pandas 一起使用，将其应用于历史记录的每个月:

#Create example dataframe
np.random.seed(1324)
idx = pd.DatetimeIndex(start='1960-01-01', periods=100, freq='d')
values = np.random.random(100)
df = pd.DataFrame(values, index=idx)

#Apply function
df.resample('M', how=spell)

我得到的是:

0
1960-01-31  1.555556
1960-02-29  1.500000
1960-03-31  1.777778
1960-04-30  6.000000

这是完美的，但是我希望能够动态地更改此函数的默认值，以便我可以将它的其他选项与 df.resample() 一起使用。我研究过 functools.partial() 但这只是针对显式设置输入参数的情况的解决方案，即。 拼写(kind='dry', how='max', threshold=0.7)。有没有一种方法可以更改函数的默认参数，使其不需要在后缀中显式设置，以便我可以将其与 df.resample() 一起使用？

Answer 1

函数的默认值存储在该函数的 func_defaults 属性中，该属性是一个值元组，与函数 的 trailing 元素配对func_code.co_varnames 元组。例如:

>>> def foo(x, y=5):
...    return x, y
...
>>> foo(10)
(10, 5)
>>> foo.func_code.co_varnames
('x', 'y')
>>> foo.func_defaults
(5,)
>>> foo.func_defaults = (7,)
>>> foo(10)
(10, 7)

你甚至可以在事后给参数一个默认值:

>>> foo.func_defaults = (2, 3)
>>> foo()
(2, 3)

---

警告:我曾想过(ab)使用mock 库来允许临时覆盖函数默认值，类似于recent answer of mine。 .但是，之后似乎将默认设置保留为 None，这意味着 mock 中存在错误(或者我误解了其行为)，或者功能困惑这样有点危险。

def foo(x=5):
    return x

assert foo() == 5
with mock.patch.object(foo, 'func_defaults', (10,)):
    assert foo() == 10

assert foo() == 5  # Oops; I'm observing foo.func_defaults to be None now

尽管如您所料，手动保存和恢复默认值似乎工作正常。

orig_defaults = foo.func_defaults
foo.func_defaults = (10,)
assert foo() == 10
foo.func_defaults = orig_defaults
assert foo() == 5