def plotOptR(n, aroundPoint=None, aroundN=None, colorStates=False, **kwargs): # don't plot red states stateArray = getStateArray(n) def filterFn(x): (iM, iD) = getNearestGridPoint(x[0], x[1]) if (stateArray[iM, iD] != g.STATE_RED): return True return False fnObj = linterp.GetLinterpFnObj(g.stateGridList, bellman.getIter(n)['opt_r']) return plot3d.plotSurface(g.grid_M, g.grid_D, fnObj, xlabel="M", ylabel="D", zlabel="opt r", colorFn=iterColorFn(n), filterFn=filterFn, drawEntireRegion=True, **kwargs)
def plotOptD(n, aroundPoint=None, aroundN=None, colorStates=False, **kwargs): fnObj = linterp.GetLinterpFnObj(g.stateGridList, bellman.getIter(n)['opt_d']) return plot3d.plotSurface(g.grid_M, g.grid_D, fnObj, xlabel="M", ylabel="D", zlabel="opt d", colorFn=iterColorFn(n), **kwargs)
