3d orbital: $|\psi|^2 \propto r^4 e^{-2r/3} \sin^4\theta \cos^2(2\phi)$
2D scatter plots
3D scatter plots
The 3D-version uses Monte-Carlo sampling + rejection sampling based on the probability density $|\psi|^2$,
so per orbital a radial distribution and angular weight is generated.