Jupyter Notebooks (Optional)#

Attention

This material was included in last year’s module NSCI0010. Complete this material if you want to refresh your memory or if you didn’t take that module.

What you will learn

After completing this worksheet, you will be able to:

  • Run and edit Jupyter Notebooks

  • Create a Jupyter Notebook from scratch

Part 1: Editing and Running Notebooks#

Step 1: Access the workshop files#

Open your NSCI0036 Cocalc project, then click on the Before_you_begin folder.

Note

Cocalc

The online platform we will be using, providing online access to virtual computers hosted in the cloud. You access Cocalc via your browser at www.cocalc.com.

Project

Every student has a Cocalc account allowing access to a Cocalc project, which is a virtual computer including operating system (Linux) and software libraries.

Jupyter Notebook

A type of file which contains Python code and formatted text, allowing us to combine computations, results and descriptive text in a single file. It is also called an IPython Notebook, and has the extension .ipynb.

Python

The programming language allowing us to perform scientific computing.

Step 2: Open the Barnsley_Fern.ipynb notebook file#

Click on the file Barnsley_Fern.ipynb to open the Jupyter Notebook file.

The file contains Python code to generate a Barnsley Fern, a fractal which simulates self-similar patterns found in nature. You don’t need to understand the mathematics, but interested students might like to read about it.

A Jupyter Notebook file is split up into cells. A cell can be of two types: A code cell which contains Python code or a markdown cell containing formatted text. We can select a cell by clicking it with the mouse, and run the selected cell by clicking the run button in the toolbar.

a

Select the first cell in the notebook and run it by clicking the run cell button a.

The second cell should now be selected. After running a cell, the next cell is selected automatically.

Run each cell in the notebook in turn by repeatedly clicking the run cell button.

Nothing will happen until you run the final cell. This is because none of the other cells generate any output! The final cell, however, should generate a plot of a green fern-like shape.

Step 3: Edit Code Cell#

Jupyter notebooks have two different modes, edit and command. To change to edit mode, press the Enter key; to return to command mode, press the Esc key.

Identify the code cell which contains the line of code npts = 50000 (about half-way down the notebook). Select the cell then press Enter to change to edit mode.

The variable npts determines the number of points the algorithm will draw.

Change the line to read npts = 500.

Return to command mode by pressing Esc.

We’d like to re-run the entire notebook. Rather than running each cell individually, run the entire notebook by pressing the ‘restart and run all’ button a. It should create a very sparse-looking fern.

Change the line back to npts = 50000 then re-run the entire notebook.

Step 4: Create a new Code Cell#

To create a new code cell, press the new cell button a. The new cell will be created immediately below the currently selected cell.

Create a new code cell below the bottom cell in the notebook

We’d like to plot another copy of the fern, this time in red.

Copy and paste the two lines of code starting plt.imshow... into the new code cell, then edit the code to change cm.Greens to cm.Reds. Run the code cell.

You should see another identical fern, this time in red.

Step 5: Delete a Cell#

To delete a cell, first select the cell then press x. (The cell must be in command mode. If the cell is in edit mode, pres Esc first).

Delete the cell you created in Step 5.

Step 6: Create a Markdown Cell#

A markdown cell contains human-readable formatted text. To create a markdown cell, first we must create a code cell then change it to a markdown cell by pressing the m key. (To change a cell from a markdown cell to a code cell, press the y key).

Create a markdown cell at the bottom of the notebook.

Markdown cells contain text and special formatting instructions. For example, text surrounded by double asterisks symbols is rendered in bold.

Change to edit mode and enter following text:
Thank you for generating the **Barnsley Fern**!

Other formatting instructions include # to denote a heading, *text* for italics and - for a bulletted list.

Instead of clicking the run button, we can use the keyboard shortcut Shift + Enter to run a cell and automatically move to the cell below.

Press Shift + Enter to render the markup cell.

Part 2: Creating a Python Notebook#

We will create a Notebook which generates a plot displaying the path of a Hurricane. It will read a time series of hurricane co-ordinates from a text file, and plot the co-ordinates on an image of the Atlantic Ocean. a

We will need to use the following two files:

atlantic-basin.png
irma.csv

Download the two files atlantic-basin.png and irma.csv to your computer.

Step 1: Navigating the File System#

The Cocalc Project contains a hierarchical file system much like an ordinary PC. To open the file browser, click the Files button on the toolbar. To navigate the file system, use the ‘Current Folder’ breadcrumb and the File List. Click on a folder in the list to open the folder, and click the breadcrumb to navigate back up the folder tree.

a

To create a new file in the current folder, type its name in the the ‘New File Name’ box, then click the arrow next to the ‘New File’ button and select the file type. To create a new folder, select ‘Folder’ at the bottom of the list of file types.

Create a new file Hurricane.ipynb in the Before_you_begin folder. (You can either type the full file name including extension .ipynb, or just type Hurricane then select ‘Jupyter Notebook’ from the list).

Step 2: Plotting Points#

Create a new code cell, paste in the following code, then execute it.

import matplotlib.pyplot as plt
x = [5, 6]
y = [6, 7]
plt.figure(figsize=(2,2))
plt.scatter(x,y)

You should see a box with two points in it.

You don’t need to fully understand the code yet, but take a look and see if you can identify what each line is doing:

  • Load the plotting code library.

  • Create two list variables containing x and y coordinates.

  • Create a figure and set its size.

  • Plot the two points (5, 6) and (6, 7).

Notice that the axis limits have been determined automatically.

Set the lower and upper axis limits to 0 and 10 by adding the following code to the bottom of the code cell:

plt.xlim(0, 10)
plt.ylim(0, 10)

Step 3: Plotting an Image#

In the next step we will display an image of the Atlantic Ocean. First we need to upload the image file to our Cocalc project. To upload a file, open the file browser, navigate to the target folder then click the ‘upload file’ button.

Upload the file atlantic-basin.png to the Before_you_begin folder.

Next, we will write code to display the image file.

Create a new code cell and paste in the following code:

import matplotlib.image as mpi

img = mpi.imread('atlantic-basin.png')
plt.imshow(img)

After running the code cell, you should see the map image within a set of axes.

Step 4: Plotting a Point on the Image#

We’d like to plot the position of New York on the map. New York has latitude/longitude co-ordinates 40.7, -74.0, but to place it on the map image we need to translate to pixel coordinates.

Create a new code cell and add the following code:

# latitude/longitude co-ordinates of edges of map
left = -90
right = -17.06
bottom = 0
top = 45

# dimensions of image in pixels
width = 964
height = 600

# lat/long of New York
lat_NY = 40.7
long_NY = -74.0

# Convert from lat/long to pixel coordinates
x_NY = (long_NY - left)*width/(right-left)
y_NY = (height*(1 - (lat_NY - bottom)/(top-bottom)))

# Print the pixel co-ordinates of New York
print("NY x pos:", x_NY)
print("NY y pos:", y_NY)

This is a lot of code, but don’t worry. You don’t understand what each line is doing.

Note

Comments

The Python interpreter ignores any text that appears after the # symbol. These lines are comments and I have added them to explain what the Python code is doing.

To add the location of New York to the map, use the scatter function as before.

In a new code cell, enter the following code

plt.imshow(img)
plt.scatter(x_NY, y_NY)
plt.text(x_NY, y_NY, "New York", color="white")

Step 5: Loading the Data#

The co-ordinates of the Hurricane are contained in the file irma.csv.

Upload the file irma.csv to the Before_you_begin folder.

Each line of the file contains data about the Hurricane at a single time point, separated by commas.

Click on the file to view the contents.

Notice that the latitude and longitude are the second and third item in each row.

Paste the following code into a new code cell.

import csv # load the code library for reading CSV files

x = []
y = []

# Open the data file
with open("irma.csv") as f:
    reader = csv.reader(f)
    # skip the first line of the file
    next(reader)
    for row in reader:
        latitude = float(row[2])
        longitude = float(row[3])

        # translate to pixel coordinates
        x.append((longitude - left)*width/(right-left))
        y.append(height*(1 - (latitude - bottom)/(top-bottom)))

print("x-coords:", x)
print("y-coords:", y)

This code opens the data file and reads the latitude and longitude from each row. It then translates each to pixel coordinates and appends the values to the two lists x and y (and don’t worry, you won’t understand this code yet).

Note

Indentation

A peculiar feature of Python is the use of indentation to separate code blocks (other languages use curly brackets { and } or begin ... end). The above example has two levels of identation, one below the with statement and one below the for statement, each idented by exacly four space characters.

Step 6: Plotting the Hurricane#

Finally, we will plot the hurricane data points on the map.

Enter the following code in a new code cell

plt.imshow(img)
plt.scatter(x, y, color='purple')

Step 7 (Optional): Animating the Plot#

The following code generates a movie displaying the movement of the Hurricane.

Enter the following code in a new cell.

import matplotlib.animation as animation

fig = plt.figure()
ax = plt.axes()
patch = plt.Circle((0, 0), 20, fc='y')


def init():
    patch.center = (20, 20)
    ax.add_patch(patch)
    return patch,

def animate(i):
    patch.center = (x[i], y[i])
    return patch,

anim = animation.FuncAnimation(fig, animate,
                               init_func=init,
                               frames=len(x),
                               interval=20,
                               blit=True)

plt.imshow(img,zorder=0)
anim.save('hurricane_irma.mp4', writer = 'ffmpeg', fps=30)

After running the code, a new file hurricane_irma.mp4 will appear in the Before_you_begin folder (be patient, the code might take a little time to run).

Click on the file hurricane_irma.mp4 to view the movie.

Step 8 (Optional Challenge): Bermuda Triangle#

Adapt the Hurricane.ipynb notebook so that it plots the vertices of the Bermuda Triangle.

  • Create a new file Bermuda_Triangle.ipynb by duplicating the Hurricane.ipynb file.

  • Create a new file bermuda.csv containing the co-ordinates of the vertices of the Bermuda Triangle. You should do this by adapting the file irma.csv. Estimate the co-ordinates of the three vertices from the Wikipedia article or elsewhere.

  • Edit the code in Bermuda_Triangle.ipynb so that it loads data from bermuda.csv