Putting a simple if-then statement on one line

Question

I'm just getting into Python and I really like the terseness of the syntax. However; is there an easier way of writing an if-then statement so it fits on one line?

For example; say I have the simple test:

if count == N:
    count = 0
else:
    count = N + 1

is there a simpler way of writing this? I mean, in Objective-C I would write this as:

count = count == N ? 0 : count + 1;

Is there something similar for python?

Edit

I know that in this instance I can use count == (count + 1) % N. I'm asking about the general syntax.

Answer 1

That's more specifically a ternary operator expression than an if-then, here's the python syntax

value_when_true if condition else value_when_false

Answer 2

count = 0 if count == N else N+1

- the ternary operator. Although I'd say your solution is more readable than this.

Answer 3

General ternary syntax:

value_true if <test> else value_false

Another way can be:

[value_false, value_true][<test>]

e.g:

count = [0,N+1][count==N]

Answer 4

<execute-test-successful-condition> if <test> else <execute-test-fail-condition>

with your code-snippet it would become,

count = 0 if count == N else N + 1

Answer 5

count = count == N ? 0 : count + 1;

Please don't do that; the next programmer to have to read your code will have a much harder time of it. "More space on screen" isn't the evil to avoid; "more time to read and understand" is the evil you should try to stomp out.