Angular Spectrum Method Python