This page was generated from source/notebooks/L4/modules.ipynb.
Binder badge
Binder badge CSC badge

Loading and using modules

What is a module?

A module in Python is simply a Python .py file that contains a list of related functions that can be loaded and used. Modules are similar to what are more generally called libraries in programming languages, which again contain code related to a specific task such as mathematical operations. There are a HUGE number of Python modules, and many of them greatly extend what can be done in a normal Python program. In fact, the abundance of free Python modules is one of the best reasons to learn and start using Python.

How can modules be loaded?

Python modules can be loaded in a number of different ways.

Loading a module

Let’s start simple with the math module. Here, we’ll load the math module using the import statement.

[1]:
import math
[2]:
math.sqrt(81)
[2]:
9.0

Here we have loaded the math module by typing import math, which tells Python to read in the functions in the math module and make them available for use. In our example, we see that we can use a function within the math library by typing the name of the module first, a period, and then the name of function we would like to use afterward (e.g., math.sqrt()). Built-in functions such as print() do not require the name of the module first since nothing is explicitly imported.

Renaming imported modules

We can also rename modules when they are imported. This can be helpful when using modules with longer names.

[3]:
import math as m
[4]:
m.sqrt(49)
[4]:
7.0
[5]:
type(m)
[5]:
module

In this example we now see that when the math module is imported, it is imported to be usable with the name m instead of math. It doesn’t matter much in our toy example here since math is not a long module name, but we will see other examples later in the course where renaming the modules is very helpful (e.g., matplotlib).

Importing a single function

It is also possible to import only a single function from a module, rather than the entire module. This is sometimes useful when using large modules that have much more available than the desired use.

[6]:
from math import sqrt
[7]:
sqrt(121)
[7]:
11.0

Though this can be useful, it has the drawback that the imported function could conflict with other built-in or imported function names, and you lose the information about which module contains the function. You should only do this when you truly need to.

Importing part of a module

Some modules have sub-modules that can also be imported without importing the entire module. We may see examples of this later when making data plots using the pyplot sub-module of the Matplotlib module. In case you’re curious, here is an example.

[8]:
import matplotlib.pyplot as plt
[9]:
plt.figure()
[9]:
<matplotlib.figure.Figure at 0x10f04c278>
<matplotlib.figure.Figure at 0x10f04c278>

This creates a new figure window for a pyplot figure. Again, we’ll see how this works and what it means later in the course.

How can modules be used?

As we see above, the easiest way to use a module is to import it an then use its functions by typing modulename.functionname() and providing the necessary arguments. Yes, it is that simple.

However, there are times you may not know the names of all of the functions in a given module, or which are part of a module. You can view the list of functions that are part of a module by using the dir() function.

[10]:
print(dir(math))
['__doc__', '__file__', '__loader__', '__name__', '__package__', '__spec__', 'acos', 'acosh', 'asin', 'asinh', 'atan', 'atan2', 'atanh', 'ceil', 'copysign', 'cos', 'cosh', 'degrees', 'e', 'erf', 'erfc', 'exp', 'expm1', 'fabs', 'factorial', 'floor', 'fmod', 'frexp', 'fsum', 'gamma', 'gcd', 'hypot', 'inf', 'isclose', 'isfinite', 'isinf', 'isnan', 'ldexp', 'lgamma', 'log', 'log10', 'log1p', 'log2', 'modf', 'nan', 'pi', 'pow', 'radians', 'sin', 'sinh', 'sqrt', 'tan', 'tanh', 'tau', 'trunc']

So that’s helpful, but what about when you don’t know what a given function does? The easiest solution is to use the help() function.

[11]:
help(math.sin)
Help on built-in function sin in module math:

sin(...)
    sin(x)

    Return the sine of x (measured in radians).

Note that you’ll need to press q to exit the help viewer.

What should I not do?

Here are a few things to avoid.

from X import *

Don’t use from X import *. This may be easier to understand by way of an example, but assuming X above is a Python module, from X import * will import all of the functions in module X. Though you might think this is helpful, it is much better to simply import X or import X as Y to keep the connection between the functions and their module. It is also much more likely you will encounter conflicting names when using from X import *.

Poor names when renaming on import

Don’t use confusing names when renaming on import. Be smart when you import modules. If you want to make the module name shorter on import, pick a reasonable abbreviation. For instance, import matplotlib as m could be confusing, especially if you’re also using import math as m in other Jupyter notebooks or script files. Similarly, import matplotlib as math is perfectly OK syntax in Python, but bound to cause a world of trouble. Remember, people need to be able to read and understand the code you write, keep it simple and logical.