目录
Numpy 创建一个通用函数
使用NumPy中的frompyfunc
函数,可以利用一个Python函数创建通用函数。
具体步骤
下面是创建一个通用函数的步骤。
- 定义Python函数。
定义一个简单的Python函数,其功能是把输入值加倍。
def double(a):
return 2 * a
- 创建通用函数。
用frompyfunc
创建通用函数时,需要指明输入参数的个数和返回对象的个数。
import numpy
def double(a):
return 2 * a
ufunc = numpy.frompyfunc(double, 1, 1)
print "Result", ufunc(numpy.arange(4))
上述代码执行后,输出如下结果。
Result [0 2 4 6]