Python代码优化