创建自己的 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. ufunc
如果你使用 np 作为 numpy 的别名):
实例
使用 if 语句检查函数是否为 ufunc:
import numpy as np
if type(np.add) == np.ufunc:
print('add is ufunc')
else:
print('add is not ufunc')
亲自试一试 »