Matrix/array operations inside a python function

Question

I am trying to use the content of a python function to make a contour plot. For example I can make:

import numpy as np
import matplotlib.mlab as mlab
import m