Need help on Python code

[ssalam21]

Hi,
Good day.
Hope you are doing well. Can you please the check the python script and tell me where I need to work to lessen the computational cost?
The link is below
https://stackoverflow.com/questions/70768384/right-method-for-finding-2-d-spatial-spectrum-from-cross-spectral-densities

Thanks