python 定义一个函数f(n)