Passing functions which have multiple return values as arguments in Python

Question

So, Python functions can return multiple values. It struck me that it would be convenient (though a bit less readable) if the following were possible.

a = [[1,2],[3,4]]

def cord():
    return 1, 1

def printa(y,x):
    print a[y][x]

printa(cord())

...but it's not. I'm aware that you can do the same thing by dumping both return values into temporary variables, but it doesn't seem as elegant. I could also rewrite the last line as "printa(cord()[0], cord()[1])", but that would execute cord() twice.

Is there an elegant, efficient way to do this? Or should I just see that quote about premature optimization and forget about this?

David Z · Accepted Answer · 2009-03-27 21:43:58Z

34

printa(*cord())

The * here is an argument expansion operator... well I forget what it's technically called, but in this context it takes a list or tuple and expands it out so the function sees each list/tuple element as a separate argument.

It's basically the reverse of the * you might use to capture all non-keyword arguments in a function definition:

def fn(*args):
    # args is now a tuple of the non-keyworded arguments
    print args

fn(1, 2, 3, 4, 5)

prints (1, 2, 3, 4, 5)

fn(*[1, 2, 3, 4, 5])

does the same.

edited Mar 27, 2009 at 21:43

answered Mar 27, 2009 at 19:56

David Z

133k29 gold badges264 silver badges284 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

regan Over a year ago

And the documentation: docs.python.org/tutorial/…

Jason Baker · Accepted Answer · 2009-03-27 19:58:47Z

6

Try this:

>>> def cord():
...     return (1, 1)
...
>>> def printa(y, x):
...     print a[y][x]
...
>>> a=[[1,2],[3,4]]
>>> printa(*cord())
4

The star basically says "use the elements of this collection as positional arguments." You can do the same with a dict for keyword arguments using two stars:

>>> a = {'a' : 2, 'b' : 3}
>>> def foo(a, b):
...    print a, b
...
>>> foo(**a)
2 3

answered Mar 27, 2009 at 19:58

Jason Baker

200k140 gold badges388 silver badges522 bronze badges

Comments

Albert Visser · Accepted Answer · 2009-03-27 22:56:10Z

3

Actually, Python doesn't really return multiple values, it returns one value which can be multiple values packed into a tuple. Which means that you need to "unpack" the returned value in order to have multiples. A statement like

x,y = cord()

does that, but directly using the return value as you did in

printa(cord())

doesn't, that's why you need to use the asterisk. Perhaps a nice term for it might be "implicit tuple unpacking" or "tuple unpacking without assignment".

answered Mar 27, 2009 at 22:56

Albert Visser

1,1346 silver badges5 bronze badges

Collectives™ on Stack Overflow

Passing functions which have multiple return values as arguments in Python

3 Answers 3

1 Comment

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

1 Comment

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related