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()
投递蚂蚁集团等公司10个岗位 >
滴滴二面52人在聊