How to Think Like a Computer Scientist: Learning with Python 2nd Edition/Recursion and exceptions

From Wikibooks, open books for an open world
Jump to: navigation, search

Recursion and exceptions[edit]

Tuples and mutability[edit]

So far, you have seen two compound types: strings, which are made up of characters; and lists, which are made up of elements of any type. One of the differences we noted is that the elements of a list can be modified, but the characters in a string cannot. In other words, strings are immutable and lists are mutable.

A tuple, like a list, is a sequence of items of any type. Unlike lists, however, tuples are immutable. Syntactically, a tuple is a comma-separated sequence of values:

Although it is not necessary, it is conventional to enclose tuples in parentheses:

To create a tuple with a single element, we have to include the final comma:

Without the comma, Python treats (5) as an integer in parentheses:

Syntax issues aside, tuples support the same sequence operations as strings and lists. The index operator selects an element from a tuple.

And the slice operator selects a range of elements.

But if we try to use item assignment to modify one of the elements of the tuple, we get an error:

Of course, even if we can't modify the elements of a tuple, we can replace it with a different tuple:

Alternatively, we could first convert it to a list, modify it, and convert it back into a tuple:

Tuple assignment[edit]

Once in a while, it is useful to swap the values of two variables. With conventional assignment statements, we have to use a temporary variable. For example, to swap a and b:

If we have to do this often, this approach becomes cumbersome. Python provides a form of tuple assignment that solves this problem neatly:

The left side is a tuple of variables; the right side is a tuple of values. Each value is assigned to its respective variable. All the expressions on the right side are evaluated before any of the assignments. This feature makes tuple assignment quite versatile.

Naturally, the number of variables on the left and the number of values on the right have to be the same:

Tuples as return values[edit]

Functions can return tuples as return values. For example, we could write a function that swaps two parameters:

Then we can assign the return value to a tuple with two variables:

In this case, there is no great advantage in making swap a function. In fact, there is a danger in trying to encapsulate swap, which is the following tempting mistake:

If we call this function like this:

then a and x are aliases for the same value. Changing x inside swap makes x refer to a different value, but it has no effect on a in __main__. Similarly, changing y has no effect on b.

This function runs without producing an error message, but it doesn't do what we intended. This is an example of a semantic error.

Pure functions and modifiers revisited[edit]

In :ref:`pure-func-mod` we discussed pure functions and modifiers as related to lists. Since tuples are immutable we can not write modifiers on them.

Here is a modifier that inserts a new value into the middle of a list:

We can run it to see that it works:

If we try to use it with a tuple, however, we get an error:

The problem is that tuples are immutable, and don't support slice assignment. A simple solution to this problem is to make insert_in_middle a pure function:

This version now works for tuples, but not for lists or strings. If we want a version that works for all sequence types, we need a way to encapsulate our value into the correct sequence type. A small helper function does the trick:

Now we can write insert_in_middle to work with each of the built-in sequence types:

The last two versions of insert_in_middle are pure functions. They don't have any side effects. Adding encapsulate and the last version of insert_in_middle to the seqtools.py module, we can test it:

The values of my_string, my_list, and my_tuple are not changed. If we want to use insert_in_middle to change them, we have to assign the value returned by the function call back to the variable:

Recursive data structures[edit]

All of the Python data types we have seen can be grouped inside lists and tuples in a variety of ways. Lists and tuples can also be nested, providing myriad possibilities for organizing data. The organization of data for the purpose of making it easier to use is called a data structure.

It's election time and we are helping to compute the votes as they come in. Votes arriving from individual wards, precincts, municipalities, counties, and states are sometimes reported as a sum total of votes and sometimes as a list of subtotals of votes. After considering how best to store the tallies, we decide to use a nested number list, which we define as follows:

A nested number list is a list whose elements are either:

  1. numbers
  2. nested number lists

Notice that the term, nested number list is used in its own definition. Recursive definitions like this are quite common in mathematics and computer science. They provide a concise and powerful way to describe recursive data structures that are partially composed of smaller and simpler instances of themselves. The definition is not circular, since at some point we will reach a list that does not have any lists as elements.

Now suppose our job is to write a function that will sum all of the values in a nested number list. Python has a built-in function which finds the sum of a sequence of numbers:

For our nested number list, however, sum will not work:

The problem is that the third element of this list, [11, 13], is itself a list, which can not be added to 1, 2, and 8.

Recursion[edit]

To sum all the numbers in our recursive nested number list we need to traverse the list, visiting each of the elements within its nested structure, adding any numeric elements to our sum, and repeating this process with any elements which are lists.

Modern programming languages generally support recursion, which means that functions can call themselves within their definitions. Thanks to recursion, the Python code needed to sum the values of a nested number list is surprisingly short:

The body of recursive_sum consists mainly of a for loop that traverses nested_num_list. If element is a numerical value (the else branch), it is simply added to sum. If element is a list, then recursive_sum is called again, with the element as an argument. The statement inside the function definition in which the function calls itself is known as the recursive call.

Recursion is truly one of the most beautiful and elegant tools in computer science.

A slightly more complicated problem is finding the largest value in our nested number list:

Doctests are included to provide examples of recursive_max at work.

The added twist to this problem is finding a numerical value for initializing largest. We can't just use nested_num_list[0], since that my be either a number or a list. To solve this problem we use a while loop that assigns largest to the first numerical value no matter how deeply it is nested.

The two examples above each have a base case which does not lead to a recursive call: the case where the element is a number and not a list. Without a base case, you have infinite recursion, and your program will not work. Python stops after reaching a maximum recursion depth and returns a runtime error.

Write the following in a file named infinite_recursion.py:

At the unix command prompt in the same directory in which you saved your program, type the following:

python infinite_recursion.py

After watching the messages flash by, you will be presented with the end of a long traceback that ends in with the following:

We would certainly never want something like this to happen to a user of one of our programs, so before finishing the recursion discussion, let's see how errors like this are handled in Python.

Exceptions[edit]

Whenever a runtime error occurs, it creates an exception. The program stops running at this point and Python prints out the traceback, which ends with the exception that occured.

For example, dividing by zero creates an exception:

So does accessing a nonexistent list item:

Or trying to make an item assignment on a tuple:

In each case, the error message on the last line has two parts: the type of error before the colon, and specifics about the error after the colon.

Sometimes we want to execute an operation that might cause an exception, but we don't want the program to stop. We can handle the exception using the try and except statements.

For example, we might prompt the user for the name of a file and then try to open it. If the file doesn't exist, we don't want the program to crash; we want to handle the exception:

The try statement executes the statements in the first block. If no exceptions occur, it ignores the except statement. If any exception occurs, it executes the statements in the except branch and then continues.

We can encapsulate this capability in a function: exists takes a filename and returns true if the file exists, false if it doesn't:

You can use multiple except blocks to handle different kinds of exceptions (see the Errors and Exceptions_ lesson from Python creator Guido van Rossum's Python Tutorial_ for a more complete discussion of exceptions).

If your program detects an error condition, you can make it raise an exception. Here is an example that gets input from the user and checks that the number is non-negative.

The raise statement takes two arguments: the exception type, and specific information about the error. ValueError is the built-in exception which most closely matches the kind of error we want to raise. The complete listing of built-in exceptions is found in section 2.3_ of the Python Library Reference_, again by Python's creator, Guido van Rossum.

If the function that called get_age handles the error, then the program can continue; otherwise, Python prints the traceback and exits:

The error message includes the exception type and the additional information you provided.

Using exception handling, we can now modify infinite_recursion.py so that it stops when it reaches the maximum recursion depth allowed:

Run this version and observe the results.

Tail recursion[edit]

When a recursive call occurs as the last line of a function definition, it is refered to as tail recursion.

Here is a version of the countdown function from chapter 6 written using tail recursion:

Any computation that can be made using iteration can also be made using recursion.

Several well known mathamatical functions are defined recursively. Factorial_, for example, is given the special operator, !, and is defined by:

0! = 1
n! = n(n-1)

We can easily code this into Python:

Another well know recursive relation in mathematics is the fibonacci sequence_, which is defined by:

fibonacci(0) = 1
fibonacci(1) = 1
fibonacci(n) = fibonacci(n-1) + fibonacci(n-2)

This can also be written easily in Python:

Both factorial and fibonacci are examples of tail recursion.

Tail recursion is considered a bad practice in languages like Python, however, since it uses more system resources than the equivalent iterative solution.

Calling factorial(1000) will exceed the maximum recursion depth. And try running fibonacci(35) and see how long it takes to complete (be patient, it will complete).

You will be asked to write an iterative version of factorial as an exercise, and we will see a better way to handle fibonacci in the next chapter.

List comprehensions[edit]

A list comprehension is a syntactic construct that enables lists to be created from other lists using a compact, mathematical syntax:

The general syntax for a list comprehension expression is:

This list expression has the same effect as:

As you can see, the list comprehension is much more compact.

Mini case study: tree[edit]

The following program implements a subset of the behavior of the Unix tree_ program.

You will be asked to explore this program in several of the exercises below.

Glossary[edit]

Exercises[edit]

#.

Run this program and describe the results. Use the results to explain why this version of swap does not work as intended. What will be the values of a and b after the call to swap?

#. Create a module named seqtools.py. Add the functions encapsulate and

insert_in_middle from the chapter. Add doctests which test that these two functions work as intended with all three sequence types.

  1. Add each of the following functions to seqtools.py:

    As usual, work on each of these one at a time until they pass all of the doctests.
  2. Write a function, recursive_min, that returns the smallest value in a nested number list:

    Your function should pass the doctests.
  3. Write a function recursive_count that returns the number of occurances of target in nested_number_list:

    As usual, your function should pass the doctests.
  4. Write a function flatten that returns a simple list of numbers containing all the values in a nested_number_list:

    Run your function to confirm that the doctests pass.
  5. Write a function named readposint that prompts the user for a positive integer and then checks the input to confirm that it meets the requirements. A sample session might look like this:

    Use Python's exception handling mechanisms in confirming that the user's input is valid.
  6. Give the Python interpreter's response to each of the following:

    #.
    #.
    #.
    #.
    You should anticipate the results before you try them in the interpreter.
  7. Use either pydoc or the on-line documentation at [http://pydoc.org http://pydoc.org]_ to find out what sys.getrecursionlimit() and sys.setrecursionlimit(n) do. Create several experiments like what was done in infinite_recursion.py to test your understanding of how these module functions work.
  8. Rewrite the factorial function using iteration instead of recursion. Call your new function with 1000 as an argument and make note of how fast it returns a value.
  9. Write a program named litter.py that creates an empty file named trash.txt in each subdirectory of a directory tree given the root of the tree as an argument (or the current directory as a default). Now write a program named cleanup.py that removes all these files. Hint: Use the tree program from the mini case study as a basis for these two recursive programs.