40from matplotlib
import cm
41import matplotlib.pyplot
as plt
42from mpl_toolkits.mplot3d
import Axes3D
46 return 1 + np.sin(x) * np.sin(y)
48def potentialSurface():
49 X = np.arange(-8, 8, 0.25)
50 Y = np.arange(-8, 8, 0.25)
51 X, Y = np.meshgrid(X, Y)
56ax = fig.gca(projection=
'3d', aspect=
'equal')
57X, Y, Z = potentialSurface()
58ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm, linewidth=0)
60x = np.loadtxt(
"vfrrt-conservative.path")
61ax.plot(x[:, 0], x[:, 1], potential(x[:, 0], x[:, 1]), color=
'b')
63x = np.loadtxt(
"trrt-conservative.path")
64ax.plot(x[:, 0], x[:, 1], potential(x[:, 0], x[:, 1]), color=
'r')
66x = np.loadtxt(
"rrtstar-conservative.path")
67ax.plot(x[:, 0], x[:, 1], potential(x[:, 0], x[:, 1]), color=
'g')