this is the python version of the original matlab code of the Diego-Restuccia, Richard-Rogerson(2008)'s paper Policy distortions and aggregate productivity with heterogeneous establishments.
The "main.ipynb" is the main driver, which will use the function in "sub_We_function.py" and "We_function.py" files during the execution.