how to convert matlab code to python code