123456789101112131415161718192021222324252627 |
- def plotIn3D(LHStok, RHStok, variables, axisRange):
- """Returns function for 3D plots
- Arguments:
- LHStok {list} -- expression tokens
- RHStok {list} -- expression tokens
- variables {list} -- variables in equation
- axisRange {list} -- axis limits
- Returns:
- graphVars {list} -- variables for plotting
- func {function} -- equation to be plotted in 3D
- """
- xmin = -axisRange[0]
- xmax = axisRange[0]
- ymin = -axisRange[1]
- ymax = axisRange[1]
- zmin = -axisRange[2]
- zmax = axisRange[2]
- meshLayers = axisRange[3]
- xrange = np.linspace(xmin, xmax, meshLayers)
- yrange = np.linspace(ymin, ymax, meshLayers)
- zrange = np.linspace(zmin, zmax, meshLayers)
- graphVars = [xrange, yrange, zrange]
-
|