Oct-20-2019, 07:30 PM
Consider the equation given in the function eq. The aim is to solve the equation for
z
(which becomes a surface in 3-space) and plot it in a 3d plot. How would one solve for values of z
over the meshgrid X,Y
? The code below only gives the values of z
along the line y=x. import numpy as np from mpl_toolkits import mplot3d from scipy.optimize import fsolve def eq(z,x,y): return x+2*y+z+np.exp(2*z)-1 X=Y=np.linspace(0,2,3) X,Y=np.meshgrid(X,Y) Z=[] for x,y in zip(X,Y): Z.append(fsolve(eq,1,args=(x,y))[0]) fig = plt.figure() ax = plt.axes(projection='3d') ax.plot_surface(X, Y, np.asarray(Z), alpha=0.5);