This cell is a markdown cell
import numpy as np
from matplotlib import pyplot
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d.axes3d import Axes3D
from pylab import *
%matplotlib inline
def f(x, y):
return 1 - cos(y)*cos(x)
y_vec = linspace(0, 2*pi, 100)
x_vec = linspace(0, 2*pi, 100)
X,Y = meshgrid(x_vec, y_vec)
Z = f(X, Y).T
fig = plt.figure(figsize=(9,6))
ax = fig.add_subplot(1, 1, 1, projection='3d')
p = ax.plot_surface(X, Y, Z, rstride=1, cstride=1,
cmap=cm.coolwarm, linewidth=0,
antialiased=False)
cb = fig.colorbar(p, shrink=0.5)