python怎么写算法