Loops

Loops#

If you want code to be executed repeatedly, you can make use of loops.

For loops#

For looping over a range of numbers, we can use a simple for loop and the range function.

In the following cell, the print(i) command will be executed a couple of times for different values of i. We iterate over a range of values:

for i in range(0, 5):
    print(i)

Note that the above code that is indented will only be excuted for the first given number (0) and continue until the last number (5) but not including it.

You can also loop over a range of numbers with a defined step, for example step 3:

for i in range(0, 10, 3):
    print(i)

Iterating over arrays allows you to do something with all array elements:

for animal in ["Dog", "Cat", "Mouse"]:
    print(animal)

Dog
Cat
Mouse

You can iterate over two arrays in parallel, pair-wise like this:

# going through arrays pair-wise
measurement_1 = [1, 9, 7, 1, 2, 8, 9, 2, 1, 7, 8]
measurement_2 = [4, 5, 5, 7, 4, 5, 4, 6, 6, 5, 4]

for m_1, m_2 in zip(measurement_1, measurement_2):
    print("Paired measurements: " + str(m_1) + " and " + str(m_2))

Paired measurements: 1 and 4
Paired measurements: 9 and 5
Paired measurements: 7 and 5
Paired measurements: 1 and 7
Paired measurements: 2 and 4
Paired measurements: 8 and 5
Paired measurements: 9 and 4
Paired measurements: 2 and 6
Paired measurements: 1 and 6
Paired measurements: 7 and 5
Paired measurements: 8 and 4

If you want to know the index of the element in the list as well, use the enumerate function:

# numbering and iterating through collections
for index, animal in enumerate(["Dog", "Cat", "Mouse"]):
    print("The animal number " + str(index) + " in the list is " + animal)

The animal number 0 in the list is Dog
The animal number 1 in the list is Cat
The animal number 2 in the list is Mouse

Generating lists in loops#

One can generate lists using for loops. The conventional way of doing this involves multiple lines of code:

# we start with an empty list
numbers = []

# and add elements
for i in range(0, 5):
    numbers.append(i * 2)
    
print(numbers)

[0, 2, 4, 6, 8]

One can also write that shorter. The underlying concept is called generators.

numbers = [i * 2 for i in range(0, 5)]

print(numbers)

[0, 2, 4, 6, 8]

The conventional combination involving an if-statements looks like this:

# we start with an empty list
numbers = []

# and add elements
for i in range(0, 5):
    # check if the number is odd
    if i % 2:
        numbers.append(i * 2)
    
print(numbers)

[2, 6]

And the short version like this:

numbers = [i * 2 for i in range(0, 5) if i % 2]

print(numbers)

[2, 6]

While loops#

Another way of looping is using the while loop. It works by checking a condition, similar to the if statement. It will interrupt execution as soon as the condition is no longer true:

number = 1024

while (number > 1):
    number = number / 2
    print(number)

Interrupting loops#

You can interrupt loops at specific points in your code using the break command:

number = 1024

while (True):
    number = number / 2
    print(number)
    
    if number < 1:
        break;

for i in range(10):
    print(i)
    if i > 5:
        break

Skipping iterations in loops#

If you want to skip iterations, you can use the continue statement. That often makes sense in combination with an if:

for i in range(0, 10):
    if i >= 3 and i <= 6:
        continue
    print(i)

Exercise 1#

Assume you have a list of filenames and you want to do something with them, for example print them out. Program a for loop which prints out all file names which end with “tif”.

file_names = ['dataset1.tif', 'dataset2.tif', 'summary.csv', 'readme.md', 'blobs.tif']

Exercise 2#

Assume you have a list of circle radii. Make a table (dictionary) with two columns: radius and area.

radii = [3, 15, 67, 33, 12, 8, 12, 9, 22]