Making Choices

Our previous lessons have shown us how to manipulate data, define our own functions, and repeat things. However, the programs we have written so far always do the same things, regardless of what data they're given. We want programs to make choices based on the values they are manipulating. To help us see what decisions they're making, we'll start by looking at how computers manipulate images.

Objectives

  • Write conditional statements including if, elif, and else branches.
  • Correctly evaluate expressions containing and and or.
  • Correctly write and interpret code containing nested loops and conditionals.
  • Explain the advantages of putting frequently-modified code in a function.

Conditionals

The tool Python gives us for doing this is called a conditional statement, and looks like this:


In [1]:
num = 37
if num > 100:
    print 'greater'
else:
    print 'not greater'
print 'done'


not greater
done

The second line of this code uses the keyword if to tell Python that we want to make a choice. If the test that follows it is true, the body of the if (i.e., the lines indented underneath it) are executed. If the test is false, the body of the else is executed instead. Only one or the other is ever executed:

Conditional statements don't have to include an else. If there isn't one, Python simply does nothing if the test is false:


In [2]:
num = 53
print 'before conditional...'
if num > 100:
    print '53 is greater than 100'
print '...after conditional'


before conditional...
...after conditional

We can also chain several tests together using elif, which is short for "else if". This makes it simple to write a function that returns the sign of a number:


In [3]:
def sign(num):
    if num > 0:
        return 1
    elif num == 0:
        return 0
    else:
        return -1

print 'sign of -3:', sign(-3)


sign of -3: -1

One important thing to notice in the code above is that we use a double equals sign == to test for equality rather than a single equals sign because the latter is used to mean assignment. This convention was inherited from C, and while many other programming languages work the same way, it does take a bit of getting used to...

We can also combine tests using and and or. and is only true if both parts are true:


In [4]:
if (1 > 0) and (-1 > 0):
    print 'both parts are true'
else:
    print 'one part is not true'


one part is not true

while or is true if either part is true:


In [5]:
if (1 < 0) or ('left' < 'right'):
    print 'at least one test is true'


at least one test is true

In this case, "either" means "either or both", not "either one or the other but not both".

Challenges

  1. True and False aren't the only values in Python that are true and false. In fact, any value can be used in an if or elif. After reading and running the code below, explain what the rule is for which values are considered true and which are considered false. (Note that if the body of a conditional is a single statement, we can write it on the same line as the if.)

    if '': print 'empty string is true'
    if 'word': print 'word is true'
    if []: print 'empty list is true'
    if [1, 2, 3]: print 'non-empty list is true'
    if 0: print 'zero is true'
    if 1: print 'one is true'
    
  2. Write a function called near that returns True if its first parameter is within 10% of its second and False otherwise. Compare your implementation with your partner's: do you return the same answer for all possible pairs of numbers?

Nesting

Another thing to realize is that if statements can be combined with loops just as easily as they can be combined with functions. For example, if we want to sum the positive numbers in a list, we can write this:


In [6]:
numbers = [-5, 3, 2, -1, 9, 6]
total = 0
for n in numbers:
    if n >= 0:
        total = total + n
print 'sum of positive values:', total


sum of positive values: 20

We could equally well calculate the positive and negative sums in a single loop:


In [7]:
pos_total = 0
neg_total = 0
for n in numbers:
    if n >= 0:
        pos_total = pos_total + n
    else:
        neg_total = neg_total + n
print 'negative and positive sums are:', neg_total, pos_total


negative and positive sums are: -6 20

We can even put one loop inside another:


In [8]:
for consonant in'bcd':
    for vowel in 'ae':
        print consonant + vowel


ba
be
ca
ce
da
de

As the diagram below shows, the inner loop runs from start to finish each time the outer loop runs once:

We can combine nesting and conditionals to implement the logic of a desired execution pattern, e.g. to access certain data, make certain calculations (e.g. on a 2-dimensional or 3-dimensional grid), etc.

Challenges

  1. Will changing the nesting of the loops in the code above—i.e., wrapping the loop over consonants around the loop over vowels —change the final result?
  1. Python (and most other languages in the C family) provides in-place operators that work like this:

    x = 1  # original value
    x += 1 # add one to x, assigning result back to x
    x *= 3 # multiply x by 3
    print x
    6
    

    Rewrite the code that sums the positive and negative numbers in a list using in-place operators. Do you think the result is more or less readable than the original?

Key Points

  • Use if condition to start a conditional statement, elif condition to provide additional tests, and else to provide a default.
  • The bodies of the branches of conditional statements must be indented.
  • Use == to test for equality.
  • X and Y is only true if both X and Y are true.
  • X or Y is true if either X or Y, or both, are true.
  • Zero, the empty string, and the empty list are considered false; all other numbers, strings, and lists are considered true.
  • Nest loops to operate on multi-dimensional data.
  • Put code whose parameters change frequently in a function, then call it with different parameter values to customize its behavior.

Next Steps

Before we go any further, we need to learn how to test whether our code is doing what we want it to do, and that will be the subject of the next lesson.