怎么用python编写这个程序