这道题求一个python代码