You can change the function by changing the fz variable (z is the complex argument). You can change the graphing bounds by changing np.linspace() bounds in x and y. n is the number of pixels per axis.

Engineers at the UCLA Samueli School of Engineering introduced a framework for universal point spread function (PSF) engineering, enabling the synthesis of arbitrary spatially varying 3D PSFs using ...

Engineers at the UCLA Samueli School of Engineering have introduced a universal framework for point spread function (PSF) engineering, enabling the synthesis of arbitrary, spatially varying 3D PSFs ...