创建你自己的通用函数
如何创建你自己的通用函数
要创建你自己的通用函数,你必须定义一个函数,就像你在 Python 中使用普通函数一样,然后使用 frompyfunc()
方法将其添加到 NumPy 通用函数库中。
frompyfunc()
方法接受以下参数
function
- 函数的名称。inputs
- 输入参数(数组)的数量。outputs
- 输出数组的数量。
示例
创建你自己的用于加法的通用函数
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]))
自己尝试 »
检查函数是否为通用函数
检查函数的 type
来检查它是否为通用函数。
通用函数应返回 <class 'numpy.ufunc'>
。
如果不是通用函数,它将返回另一种类型,例如此用于连接两个或多个数组的内置 NumPy 函数
如果根本无法识别该函数,则会返回错误
要在 if 语句中测试函数是否为通用函数,请使用 numpy.ufunc
值(如果使用 np 作为 numpy 的别名,则使用 np.ufunc
)
示例
使用 if 语句检查函数是否为通用函数
import numpy as np
if type(np.add) == np.ufunc
print('add 是通用函数')
else
print('add 不是通用函数')
自己尝试 »