03 Introduction to Python III — Functions

We are continuing from the previous lesson in the "work directory" ~/PHY494/03_python. We will use ipython and your text editor.

Functions

Functions package re-useable code and generalize code by allowing variable inputs. Writing code with functions is essential for debugging and reusability. If you have, say, 10 lines of code (or even less) that do something specific, consider making it a function.

Basic structure a of a python function

def func_name(arg1, arg2, ...):
    """documentation string (optional)"""
    # body
    ...
    return results

Repackage our step function:

# Heaviside step function

def heaviside(x):
   """Heaviside step function"""

   theta = None
   if x < 0:
      theta = 0.
   elif x == 0:
      theta = 0.5
   else:
      theta = 1.

   return theta

When you run it… then nothing happens: The function was defined but not called.

Now add at the end of heaviside.py

x = 3
theta = heaviside(x)

print("Theta(" + str(x) + ") = " + str(theta))

and run it. Now the function is being called (theta = heaviside(x)) and its return value assigned to the variable theta.

Activity: create functions

1. create a file myfuncs.py
2. add heaviside() to the file
3. add a function fahrenheit2kelvin() to convert from Fahrenheit to Kelvin ¹
4. add a function kelvin2celsius() to convert from Kelvin to Celsius (subtract 273.15).

Try out your functions (in ipython):

%run myfuncs.py

heaviside(5)
fahrenheit2kelvin(100)
kelvin2celsius(300)

kelvin2celsius(fahrenheit2kelvin(100))

should give ²

1.0
310.92777777777775
26.850000000000023
37.77777777777777

Activity: Plotting the step function

Perform this activity using pair programming³.

Use what you learnt about loops and functions to plot the Heaviside step function. Create a program step_plot.py that

1. defines the Heaviside step function $\Theta(x)$ (use heaviside(x) from above);
2. generates a list of $x$ values [-4, -3.5, -3, ..., 0, 0.5, 1, 1.5, ... 4];
3. evaluates $\Theta(x)$ for all $x$ values and stores the results in a list;
4. prints the lists of $x$ and $\Theta(x)$ values; it should look like

    -4.0 0.0
    -3.5 0.0
    ...
    3.5 1.0
    4.0 1.0 
   

5. BONUS: plots $\Theta(x)$ against $x$; see the Basic Plotting example, namely you can use code like

   import matplotlib.pyplot as plt
   plt.plot(xvalues, thetas, '-o', color="red", linewidth=2)
   plt.show()
   

Does your graph look the way that you expect it? ⁴

Footnotes