Python Basics – Control Flow

Programs that only assign variables and print out strings are, well, boring! Things get more interesting when you have to make choices, performing a certain action based on some condition, another action based on another. This guide will show you how to make branching decisions, how to conditionally execute code, as well as how to use common looping constructs.

The simplest branching decision is comparing one thing to another. In Python, the if statement evaluates a condition and executes the block which follows it only when the condition is met.

In the example above, we are asking that Python should print '1 is 1' if the condition 1 == 1 is true. When you run the statements above, the condition is in fact met, and the print statement would execute.

The if statement is of the general form:

Where the if keyword is followed by a condition, the condition is followed by a colon (:) and a newline, then an indented block of statements to be executed when the condition is met. The condition can be any logical condition, as long as it can be evaluated for truthfulness - more on that in a bit.

You may have noticed that the condition is "bare", not surrounded by parenthesis. Python was designed to be terse and clutter-free - eschewing braces, parenthesis and other enclosing symbols for expressions unless absolutely necessary. The if condition must be terminated with a colon, which is the clue for the Python parser to know that the condition portion is ended.

What follows the colon is a line-break and an indented block. Python uses white-space to distinguish code blocks. You can use spaces or tabs to create a block. When several statements use the same indentation, they are considered a block. Python in fact insists that separate statements use the same indentation in a block. It would complain if you forget to indent when a block is expected, as well as if you use varying indentations. This example fails to indent after the if statement:

This example over-indents the second print statement:

While this example under-indents the second print statement:

Most modern IDEs (Integrated Development Environment) help you keep indentation using tabs or spaces, but since whitespace characters are not visible in some editors, you should take care to properly indent blocks - the source of oh-too-many programming errors.

For short statements, you can also write a single line condition:

A condition is considered true when the interpreter evaluates it and finds it to either be True or have a non-None (null in other languages) value. The following conditions are all true, and will produce a printed response:

Python conditions are not met when the expression evaluates to False or is None. The following conditions will not be met, and result in nothing being printed:

In some languages such as JavaScript you may test a "variable" for existence. Python does not have such a built in notion. The following if statement will produce an error unless the variable thing was previously created:

The error produced here is not particular to the if statement, but it does illustrate what happens when the condition produces an error rather than a true or false value.

More complex branching can be produced by using the else clause in conjunction with if:

The branching decision above executes the second block, printing out '1 is NOT less than 1' because the if condition is not met. Note that the else keyword is indented same as the if keyword, and followed by a semicolon. Using else complements if in executing the respective code block when the if condition is not met.

To check another condition in case the first one was not met, use the elif keyword instead of else, followed by its own condition:

Above, we see that the second condition 1 <= 1 is met. elif is only evaluated if no preceding condition is met. The following will print "true", not "true too" because the first condition is met, so the elif condition is skipped and not evaluated:

A final else: at the end of a chain of if and elif would catch any remaining condition not met by any preceding evaluation:

Sometime we want to execute code repeatedly until a condition is met, or while some condition remains true.

The input() function prompts the user to enter some text. The code above waits for the user to type in q+ENTER in the terminal. Any other character typed will cause it to repeatedly wait for input again. The while condition is always True here, so it seems this would go on forever. But if the user enters q, the if condition would be met and the break would execute, exiting the while loop. The break statement causes the current loop to exit the block immediately and halt the loop.

As before, the while construct expects a colon : after the condition. As long as the condition is met, the block following the condition would execute. It would continue executing until either the condition evaluates to a none-true value (False or None), or until a break statement is encountered in the block.

It is very common to need to loop a certain number of times, and know which round you are executing. Let's count to three, for example:

The code assigns the variable x 1 to start with, and with each iteration of the while loop, x is incremented by 1. When the condition x < 4 is no longer true - when x is 4 - the loop exits.

For such tasks where a variable holds a value and is then incremented each iteration, there's a cleaner syntax: the for loop. The Python for loop is most commonly used to assign a variable a sequence of values until the sequence is exhausted. Here is counting again - this time with a for loop:

This time, we don't need to explicitly bump up x each iteration - the for loop does that for us. The range() function produces a sequence of numbers from 1 to 3. The full signature of the range function is range(_start_,_stop_,_step_). In our case above, the loop exits when x becomes 4. The stop argument to range() is exclusive: it would not be assigned and the block would not execute once the range reached that value.

Using the step argument lets you bump the value by a value other than one. Here is how we can print some even numbers by starting at zero and incrementing by two:

You can count backwards, setting the step to a negative number:

There is also a shorthand if you want a numeric sequence from zero to some value, using the simplified signature range(_stop_)

This simplified range always starts at zero, and terminates with the number below the given stop value. Our example was range(3) so the last number printed in the loop is 2.

The for loop syntax uses the in keyword after the variable name. This should clue us to the fact that the range is some kind of sequence - not necessarily a numeric one! For example, a string is a sequence of characters. We can loop over those characters like so:

As you can see, each letter from the string 'Hi' is assigned to the variable thing in each iteration of the loop and given to the print() statement in the block.

String is just one of the sequence types you can provide the for loop. Other types include sets, lists, tuples, and range as we saw.

Here's a way to loop over a list of strings:

The variable stooges is assigned a list of strings. The for loop iterates over the list, assigning the variable stooge one name at the time from the stooges list (in order) and passes it along to the inner block which prints it out.

As with the while loop, you may stop iterating through a sequence using the break statement:

The loop above broke out of the block when the if condition found that the current value is 'far', so the loop stopped processing before the last item in the sequence got to 'too far'.

Sometimes, it is useful to "skip" a value in a sequence. The keyword continue lets you do just that. Using continue in a block skips execution of any remaining statements in the block, but lets the loop continue with the next value in the sequence.

The loop above continued iterating through all three values in the list of fruits. When the value 'bad apple' was assigned to the fruit variable, the condition 'bad' in fruit was met, so the rest of the block was skipped for that iteration only - the print statement did not execute that time. So the result is that all values except 'bad apple' where printed.

Branching and looping are core programming tasks that most of us will need to perform on a regular basis. This guide explained how to execute code conditionally using the if statement. You also learned to use the while and for looping constructs in order to repeat execution of a code block, and how to stop such loop execution midway. Using the techniques and concepts from this guide will set you on your way for more advanced and complex programming in Python.