Python: Могу ли я частично применить приёмник с помощью инициализатора?

initializer приходит после iterable, Это вызывает проблемы для частичного применения. Рассмотрим эти (тривиальные) примеры:

In [1]: from functools import reduce, partial

In [2]: f = partial(reduce, lambda a,b: a+b, 100)

In [3]: f([1,2,3])
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-3-816cce84b257> in <module>()
----> 1 f([1,2,3])

TypeError: reduce() arg 2 must support iteration

In [4]: f = partial(reduce, lambda a,b: a+b, initializer=100)

In [5]: f([1,2,3])
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-5-816cce84b257> in <module>()
----> 1 f([1,2,3])

TypeError: reduce() takes no keyword arguments

Есть ли какая-то хитрость, чтобы обойти это?

2 ответа

Решение

Просто измените порядок аргументов:

>>> f = partial(lambda func, init, iterable: reduce(func, iterable, init), 
                lambda a,b: a+b, 100)
>>> f([1,2,3])
106

Марат дал отличное решение. Ради интереса я написал foldl с аргументами перевернул. Мне любопытно, как это можно сравнить с производительностью с использованием лямбды, чтобы перевернуть аргументы, как это сделал Марат.

def foldl(f: Callable, acc: Any, xs: Iterable) -> Any:
    if not xs:
        return acc
    return foldl(f, f(acc, xs[0]), xs[1:]) 
Другие вопросы по тегам