如何将整数系列作为参数传递给python函数？

Question

Is there a way to execute the following statement in a more concise way in python?

instead of

a, b, c, d = f(1), f(2), f(3), f(4)

this:

a, b, c, d = some_way(f(x))

Answer 1

You can try map() :

>>> def f(x): return x*x
... 
>>> a,b,c,d = map(f, (1,2,3,4))
>>> 
>>> a
1
>>> b
4
>>> c
9
>>> d
16

If your function arguments will always be consecutive, then you can also do:

>>> a,b,c,d = map(f, range(1,5))

Answer 2

You can use list comprehension like this

a, b, c, d = [f(i) for i in xrange(1, 5)]

Or using map function, like this

a, b, c, d = map(f, xrange(1, 5))

In both the cases,

print a, b, c, d

will print

1 4 9 16

Edit:

As mentioned in the comments section, here is the curried version

def curry_function(function, first, second):
    return lambda third: function(first, second, third)

f = curry_function(f, "dummy1", "dummy2")

And then you can use the code shown above. Instead of writing our own version of currying function, we can use functools.partial like this

from functools import partial
f = partial(f, "dummy1", "dummy2")

Answer 3

Just an additional comment that requires some code... in your case where you want to supply arguments, if you use a list comprehension, you don't need the curry function:

# f(arg1, arg2, index)

a,b,c,d = [f(arg1, arg2, x) for x in range(1,5)]

List comprehensions are mostly equivalent in use to map, but this is one way in which they are superior.