创建您自己的 ufunc
如何创建您自己的 ufunc
要创建您自己的 ufunc，您需要像在 Python 中定义普通函数一样定义一个函数，然后使用 frompyfunc() 方法将其添加到您的 NumPy ufunc 库中。
frompyfunc() 方法采用以下参数
function- 函数的名称。inputs- 输入参数（数组）的数量。outputs- 输出数组的数量。
示例
创建您自己的加法 ufunc
import numpy as np
def myadd(x, y)
return x+y
myadd = np.frompyfunc(myadd, 2, 1)
print(myadd([1, 2, 3, 4], [5, 6, 7, 8]))
自己动手试一试 »
检查一个函数是否为 ufunc
检查函数的类型以确定它是否为 ufunc。
ufunc 应返回 <class 'numpy.ufunc'>。
如果它不是 ufunc，它将返回另一种类型，例如这个内置的 NumPy 函数，用于连接两个或多个数组
如果根本无法识别该函数，则会产生错误
要在 if 语句中测试函数是否为 ufunc，请使用 numpy.ufunc 值（如果将 np 用作 numpy 的别名，则为 np.ufunc）
示例
使用 if 语句检查函数是否为 ufunc
import numpy as np
if type(np.add) == np.ufunc
print('add 是 ufunc')
else
print('add 不是 ufunc')
自己动手试一试 »
