In [1]:
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
using meshgrid to compute functions on a 2d grid¶
In [2]:
xlin = np.arange(-5,5,0.1) # 1d array
ylin = np.arange(-5,5,0.1) # 1d array
xm,ym = np.meshgrid(xlin,ylin) # 2d arrays
In [5]:
xm.shape
Out[5]:
In [3]:
plt.imshow(ym)
Out[3]:
In [4]:
z = np.sin(xm**2 + 0.5*ym**2)
In [6]:
plt.imshow(z)
plt.colorbar()
Out[6]:
In [7]:
plt.contourf(xlin,ylin,z)
plt.colorbar()
Out[7]:
In [ ]: