Python数值计算（2）——函数的定义和使用

本次讨论一下函数的定义和使用。

1. 定义函数

在Python中，定义函数是一件是否简单的事情，例如我们要定义一个三次函数:

则代码如下：

def f(x):
    return (x-1)*(x-2)*(x-3)

2. 函数的使用

通过函数计算单个值，和Python中调用其他函数没有什么区别，例如上面的函数，要计算，代码如下：

print(f(4)) # 6
print(f(-1)) # -24

但是，更多的时候，我们需要计算的是一组x所对应的值，这个时候通过for循环之类的遍历可以完成工作，不过可以充分利用python的语法特性，使用map函数，实现映射（函数本质上就是一种映射）：

x=list(range(-2,5))
print(x) # [-2, -1, 0, 1, 2, 3, 4]
y=list(map(f,x))
print(y) # [-60, -24, -6, 0, 0, 0, 6]

注意：range函数返回的是一个range对象，map函数返回的是一个map对象，不能直接使用，不过可以提供给list构造列表实例。

不过，在后面使用到NumPy以后，我们会使用另外一种方式，即numpy.vectorize方法，后面将会介绍。