python代码优化