Numpy 创建一个通用函数

使用NumPy中的frompyfunc函数，可以利用一个Python函数创建通用函数。

具体步骤

下面是创建一个通用函数的步骤。

1. 定义Python函数。

定义一个简单的Python函数，其功能是把输入值加倍。

def double(a):
    return 2 * a 

2. 创建通用函数。

用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]

攻略小结

我们定义了一个Python函数，其功能是把输入值加倍。实际上把字符串作为这个函数的输入值也是可以的，Python语法允许这样做。我们用NumPy中的frompyfunc函数，基于这个Python函数创建了一个通用函数。