Python "карта" и распаковка аргументов

Я знаю, что

map(function, arguments)

эквивалентно

for argument in arguments:
    function(argument)

можно ли использовать функцию Карты, чтобы сделать следующую?

for arg, kwargs in arguments:
    function(arg, **kwargs)
18
map-function python

4 ответов

вы можете с лямбда:

map(lambda a: function(a[0], **a[1]), arguments)

или вы можете использовать выражение генератора или понимание списка, в зависимости от того, что вы хотите:

(function(a, **k) for a, k in arguments)
[function(a, **k) for a, k in arguments]

В Python 2, map() возвращает список (поэтому понимание списка эквивалентно), в Python 3,map() является генератором (поэтому выражение генератора может заменить его).

нет встроенного или стандартного метода библиотеки, который делает это напрямую; случай использования слишком специализирован.

17

только для позиционных аргументов вы можете использовать itertools.starmap(fun, args):

возвращает итератор, значения которого возвращаются из функции, вычисленной с помощью кортежа аргументов, взятых из данной последовательности.

пример:

from itertools import starmap

def f(i, arg):
    print(arg * (i+1))

for _ in starmap(f, enumerate(["a", "b", "c"])):
    pass

принты:

a
bb
ccc
9

вы просто должны помнить, что map передаст аргументы функции как один Кортеж, а не отдельные аргументы. Если вы не можете изменить исходную функцию, вы можете вызвать ее с помощью вспомогательной функции:

def f(tup):
    args, kwargs = tup
    function(args, **kwargs)

map(f, arguments)
0

Я продолжал работать в той же необходимости и в конечном итоге сделал следующую функцию: 

def kwarg_map(element_constructor, **kwarg_lists):
    """
    A helper function for when you want to construct a chain of objects with individual arguments for each one.  Can
    be easier to read than a list expansion.

    :param element_constructor: A function of the form object = fcn(**kwargs)
    :param kwarg_lists: A dict of lists, where the index identifies two which element its corresponding value will go.
    :return: A list of objects.

    e.g. Initializing a chain of layers:
        layer_sizes = [784, 240, 240, 10]
        layers = kwarg_map(
            Layer,
            n_in = layer_sizes[:-1],
            n_out = layer_sizes[1:],
            activation = ['tanh', 'tanh', 'softmax'],
            )

    is equivalent to:
        layers = [Layer(n_in=784, n_out=240, activation='tanh'), Layer(n_in=240, n_out=240, activation='tanh'), Layer(n_in=240, n_out=10, activation='softmax')]
    """
    all_lens = [len(v) for v in kwarg_lists.values()]
    assert len(kwarg_lists)>0, "You need to specify at least list of arguments (otherwise you don't need this function)"
    n_elements = all_lens[0]
    assert all(n_elements == le for le in all_lens), 'Inconsistent lengths: %s' % (all_lens, )
    return [element_constructor(**{k: v[i] for k, v in kwarg_lists.iteritems()}) for i in xrange(n_elements)]
0