Meshgrid and contours

A common task is a graphical representation of a scalar function over a rectangle:

Meshgrid and contours

For this, first we have to generate a grid on the rectangle [a,b] x [c,d]. This is done using the meshgrid command:

n = ... # number of discretization points along the x-axis
m = ... # number of discretization points along the x-axis 
X,Y = meshgrid(linspace(a,b,n), linspace(c,d,m))

X and Y are arrays with (n,m) shape such that Meshgrid and contours contains the coordinates of the grid point as shown in the next figure (Figure 6.6):

Figure 6.6: A rectangle discretized ...

Get Scientific Computing with Python 3 now with O’Reilly online learning.

O’Reilly members experience live online training, plus books, videos, and digital content from 200+ publishers.