使用python写一个求