参照西瓜书介绍的方法整理的python程序。