python中numpy自定义函数

在numpy中，可以使用frompyfunc()函数自定义函数。该函数接受四个参数：自定义函数、输入参数数量、输出参数数量和返回值类型。例如，下面的代码定义了一个将两个数相加的自定义函数：

import numpy as np

def add(x, y):
    return x + y

add_func = np.frompyfunc(add, 2, 1)

a = np.array([1, 2, 3])
b = np.array([4,5, 6])
c = add_func(a, b)

print(c)

输出结果为：

[5 7 9]