gw_blob-2.py

from kerrgeodesic_gw import surface_density_toy_model
param = [6.5, 0, 0.6, 0.1]
s_plot = lambda r, phi: surface_density_toy_model(r, phi, param)
r, phi, z = var('r phi z')
g = plot3d(s_plot, (r, 6, 8), (phi, -0.4, 0.4), \
           transformation=(r*cos(phi), r*sin(phi), z))
sphinx_plot(g)