把matlab code转为 python 的代码 求