1
from mpl_toolkits.mplot3d import Axes3D
2
from matplotlib import cm
3
import matplotlib.pyplot as plt
10
from deap import benchmarks
12
def rastrigin_arg0(sol):
13
return benchmarks.rastrigin(sol)[0]
16
ax = Axes3D(fig, azim = -29, elev = 50)
17
X = np.arange(-5, 5, 0.1)
18
Y = np.arange(-5, 5, 0.1)
19
X, Y = np.meshgrid(X, Y)
22
for i in xrange(X.shape[0]):
23
for j in xrange(X.shape[1]):
24
Z[i,j] = rastrigin_arg0((X[i,j],Y[i,j]))
26
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.jet, linewidth=0.2)
b'\\ No newline at end of file'