np.frompyfunc()函数的使用 Numpy | 函数向量化

python自定义函数并调用：

def func_x(a):
	y = a + 1
	return y

print(func_x(1)) # 调用
# 结果：
# 2

但是如果a不是一个数，而是一个向量/数组呢？这时候就需要借助numpy中的通用函数np.frompyfunc(func, 1, 1)
这里的参数func指的是你要使用的函数，第二个参数为func中的参数个数，第三个参数为func中的返回值的个数
假设A = [0, 1, 2]

def func_x(a):
	y = a + 1
	return y
A = [0, 1, 2]
func_ = np.frompyfunc(func_x, 1, 1)
y = func_(A)
print(y)
# 结果：
# [1 2 3]

参考博客：函数向量化运算的多种方法function/lambda + numpy.frompyfunc()/map和numpy.vectorize()