ans函数python