编写一个python程序