168

Possible Duplicate:
Python Ternary Operator
Putting a simple if-then statement on one line

Is there a way to compress an if/else statement to one line in Python?
I oftentimes see all sorts of shortcuts and suspect it can apply here too.

AnovaVariance
  • 1,875
  • 2
  • 11
  • 6

4 Answers4

277

An example of Python's way of doing "ternary" expressions:

i = 5 if a > 7 else 0

translates into

if a > 7:
   i = 5
else:
   i = 0

This actually comes in handy when using list comprehensions, or sometimes in return statements, otherwise I'm not sure it helps that much in creating readable code.

The readability issue was discussed at length in this recent SO question better way than using if-else statement in python.

It also contains various other clever (and somewhat obfuscated) ways to accomplish the same task. It's worth a read just based on those posts.

Community
  • 1
  • 1
Levon
  • 118,296
  • 31
  • 184
  • 178
  • How do i use it in the return statement? `return count if (count * (distance < within_radius)) > 0 else (continue)` – technazi Oct 31 '18 at 23:38
64

Python's if can be used as a ternary operator:

>>> 'true' if True else 'false'
'true'
>>> 'true' if False else 'false'
'false'
sloth
  • 91,747
  • 17
  • 156
  • 204
  • Just want to add the "true" command-line one-line syntax for the that. I have searched a lot how to do that, but no one answer contains this information: $ python -c "import sys; var=1; [ sys.stdout.write('POS\n') if var>0 else sys.stdout.write('NEG\n')]" – Alexander Samoylov Oct 08 '18 at 10:38
34

Only for using as a value:

x = 3 if a==2 else 0

or

return 3 if a==2 else 0
f p
  • 2,937
  • 1
  • 24
  • 33
25

There is the conditional expression:

a if cond else b

but this is an expression, not a statement.

In if statements, the if (or elif or else) can be written on the same line as the body of the block if the block is just one like:

if something: somefunc()
else: otherfunc()

but this is discouraged as a matter of formatting-style.

BrenBarn
  • 210,788
  • 30
  • 364
  • 352