用python怎么编写这个程序