from sympy import *
from spb import *
x, y, a, b, d = symbols("x y a b d")
plot3d(
    cos(x**2 + y**2) * exp(-(x**2 + y**2) * d),
    prange(x, -2*a, 2*a), prange(y, -2*b, 2*b),
    params={
        a: (1, 0, 3),
        b: (1, 0, 3),
        d: (0.25, 0, 1),
    },
    backend=PB, use_cm=True, n=100, aspect=dict(x=1.5, y=1.5, z=0.75),
    wireframe=True, wf_n1=15, wf_n2=15, throttled=True)