return value in Python

Question

I would like my function to return 1, 2, or 3 values, in regard to the value of bret and cret. How to do something more clever than this ? :

def myfunc (a, bret=False, cret=False):
  b=0
  c=0
  if bret and cret:
   return a,b,c
  elif bret and not cret:
   return a,b
  elif not bret and cret:
   return a,c
  else:
   return a

I'm not sure you want that. What if next question will be `how to handle tuples with different lengths`? — alko, Dec 07 '13 at 13:58
@alko Yup - especially for the two-form return where whether it's b or c is lost... — Jon Clements, Dec 07 '13 at 14:01
this is just an example @aro, in my real code `b` and `c` are more interesting values... — Basj, Dec 07 '13 at 16:14

score 4 · Accepted Answer · answered Dec 07 '13 at 13:50

4

How about:

def myfunc(a, bret=False, cret=False):
   ...
   return (a,) + ((b,) if bret else ()) + ((c,) if cret else ())

answered Dec 07 '13 at 13:50

NPE

486,780
108
951
1,012

score 3 · Answer 2 · answered Dec 07 '13 at 13:46

3

def myfunc (a, bret=False, cret=False):
  b, c, ret = 0, 0, [a]
  if bret: ret.append(b)
  if cret: ret.append(c)
  return ret

answered Dec 07 '13 at 13:46

thefourtheye

233,700
52
457
497

Robert Kajic · Answer 3 · 2013-12-07T14:00:14.143

2

def myfunc(a, bret=False, cret=False):
    return [v for v, ret in [(a, True), (0, bret), (0, cret)] if ret]

Take a moment to read about Python List Comprehensions. They are handy very often!

edited Dec 07 '13 at 14:00

answered Dec 07 '13 at 13:53

Robert Kajic

8,689
4
44
43

score 1 · Answer 4 · edited May 23 '17 at 12:16

1

You might simply append your return values to a list or use yield. You can iterate over the result afterwards and do what you please. Here's an explanation to the yield-Keyword: What does the "yield" keyword do in Python?

edited May 23 '17 at 12:16

Community

1
1

answered Dec 07 '13 at 13:48

TheWormKill

23
1
6

ddetoni · Answer 5 · 2013-12-07T14:03:03.127

1

You could return a list. Look:

def myfunc (a, bret=False, cret=False):

    list = []
    list.append(a)
    b=0
    c=0

    if bret:
        list.append(b)

    if cret:
        list.append(c)

    return list

Solve your question?

edited Dec 07 '13 at 14:03

answered Dec 07 '13 at 13:48

ddetoni

33
1
7

no, the problem is not having a list. The problem is that this won't work anymore if 3 conditions, then there would be 2^3=8 cases to distinguish ! not beautiful ;) – Basj Dec 07 '13 at 13:50
This is equivalent to a much earlier solution posted by @thefourtheye (and I'd recommend against calling the variable `list` since this shadows the built-in function.) – NPE Dec 07 '13 at 14:27

Arovit · Answer 6 · 2013-12-07T14:32:58.700

1

Please note if you have to return tuple (in your question it returns tuple) use this -

def myfunc(a, bret=False, cret=False):
    return tuple([a] + [0 for i in (bret, cret) if i])

edited Dec 07 '13 at 14:32

answered Dec 07 '13 at 13:56

Arovit

3,579
5
20
24

1

This returns `bret`/`cret` instead of `b`/`c`. – NPE Dec 07 '13 at 13:57
Return value is always - a, a,0 or a,0,0 – Arovit Dec 07 '13 at 14:33

return value in Python

6 Answers6