Repeating Actions with Loops

Overview

Teaching: 30 min
Exercises: 0 min

Questions

• How can I do the same operations on many different values?

Objectives

• Explain what a for loop does.

• Correctly write for loops to repeat simple calculations.

• Trace changes to a loop variable as the loop runs.

• Trace changes to other variables as they are updated by a for loop.

• previously, we made some plots for inflammation-01.csv
• we have lots more files
• create plots for all of them

• repeat the plotting but not manually!

• first example: access numbers in a list

odds = [1, 3, 5, 7]

• four print statements:

print(odds[0])
print(odds[1])
print(odds[2])
print(odds[3])

1
3
5
7

This is a bad approach for three reasons:

1. Not scalable. Imagine you need to print a list that has hundreds of elements. It might be easier to type them in manually.

2. Difficult to maintain. If we want to decorate each printed element with an asterisk or any other character, we would have to change four lines of code. While this might not be a problem for small lists, it would definitely be a problem for longer ones.

3. Fragile. If we use it with a list that has more elements than what we initially envisioned, it will only display part of the list’s elements. A shorter list, on the other hand, will cause an error because it will be trying to display elements of the list that do not exist.

odds = [1, 3, 5]
print(odds[0])
print(odds[1])
print(odds[2])
print(odds[3])

1
3
5

---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-3-7974b6cdaf14> in <module>()
      3 print(odds[1])
      4 print(odds[2])
----> 5 print(odds[3])

IndexError: list index out of range

• better: for loop
• similar syntax as yesterday with the shell but without do and done
• beware the indentation!!!

odds = [1, 3, 5, 7]
for num in odds:
    print(num)

1
3
5
7

• shorter and more robust!

odds = [1, 3, 5, 7, 9, 11]
for num in odds:
    print(num)

1
3
5
7
9
11

The improved version uses a for loop to repeat an operation — in this case, printing — once for each thing in a sequence. The general form of a loop is:

for variable in collection:
    # do things using variable, such as print

Using the odds example above, the loop might look like this:

• loop variable can be called anything we like
• mus end in a colon
• indent
• loop body stops where the indentation stops

EXERCISE: Understanding the loops

Given the following loop:

word = 'oxygen'
for char in word:
    print(char)

How many times is the body of the loop executed?

• 3 times
• 4 times
• 5 times
• 6 times

Solution

The body of the loop is executed 6 times.

What’s in a name?

In the example above, the loop variable was given the name num as a mnemonic; it is short for ‘number’. We can choose any name we want for variables. We might just as easily have chosen the name banana for the loop variable, as long as we use the same name when we invoke the variable inside the loop:

odds = [1, 3, 5, 7, 9, 11]
for banana in odds:
    print(banana)

1
3
5
7
9
11

It is a good idea to choose variable names that are meaningful, otherwise it would be more difficult to understand what the loop is doing.

Here’s another loop that repeatedly updates a variable:

length = 0
names = ['Curie', 'Darwin', 'Turing']
for value in names:
    length = length + 1
print('There are', length, 'names in the list.')

There are 3 names in the list.

• trace the execution of the loop step by step
• Note that a loop variable is a variable that is being used to record progress in a loop. It still exists after the loop is over, and we can re-use variables previously defined as loop variables as well:

name = 'Rosalind'
for name in ['Curie', 'Darwin', 'Turing']:
    print(name)
print('after the loop, name is', name)

Curie
Darwin
Turing
after the loop, name is Turing

• this allows us to find the length
• there is also a built-in method, since this is such a common operation len:

print(len([0, 1, 2, 3]))

4

• len is much faster than any function we could write ourselves,
• easier to read
• works for many other things as well that you will meet in the future

EXERCISE: From 1 to N

Python has a built-in function called range that generates a sequence of numbers. range can accept 1, 2, or 3 parameters.

• If one parameter is given, range generates a sequence of that length, starting at zero and incrementing by 1. For example, range(3) produces the numbers 0, 1, 2.
• If two parameters are given, range starts at the first and ends just before the second, incrementing by one. For example, range(2, 5) produces 2, 3, 4.
• If range is given 3 parameters, it starts at the first one, ends just before the second one, and increments by the third one. For example, range(3, 10, 2) produces 3, 5, 7, 9.

Using range, write a loop that uses range to print the first 3 natural numbers:

1
2
3

Solution

for number in range(1, 4):
    print(number)

EXERCISE: Summing a list

Write a loop that calculates the sum of elements in a list by adding each element and printing the final value, so [124, 402, 36] prints 562

Solution

numbers = [124, 402, 36]
summed = 0
for num in numbers:
    summed = summed + num
print(summed)

INFO: Computing Powers With Loops

Exponentiation is built into Python:

print(5 ** 3)

125

Write a loop that calculates the same result as 5 ** 3 using multiplication (and without exponentiation).

Solution

result = 1
for number in range(0, 3):
    result = result * 5
print(result)

INFO: Computing the Value of a Polynomial

The built-in function enumerate takes a sequence (e.g. a list) and generates a new sequence of the same length. Each element of the new sequence is a pair composed of the index (0, 1, 2,…) and the value from the original sequence:

for idx, val in enumerate(a_list):
    # Do something using idx and val

The code above loops through a_list, assigning the index to idx and the value to val.

Suppose you have encoded a polynomial as a list of coefficients in the following way: the first element is the constant term, the second element is the coefficient of the linear term, the third is the coefficient of the quadratic term, etc.

x = 5
coefs = [2, 4, 3]
y = coefs[0] * x**0 + coefs[1] * x**1 + coefs[2] * x**2
print(y)

97

Write a loop using enumerate(coefs) which computes the value y of any polynomial, given x and coefs.

Solution

y = 0
for idx, coef in enumerate(coefs):
    y = y + coef * x**idx

Key Points

• Use for variable in sequence to process the elements of a sequence one at a time.

• The body of a for loop must be indented.

• Use len(thing) to determine the length of something that contains other values.