在 NumPy 中创建自己的通用函数

Numpy 中的通用函数是简单的数学函数。这只是我们在 Numpy 库中赋予数学函数的一个术语。 Numpy 提供了涵盖各种操作的各种通用函数。但是，我们可以在Python中创建自己的通用函数。要在 NumPy 中创建自己的通用函数，我们必须应用以下给出的一些步骤：

通常使用 def 关键字定义函数。
使用 frompyfunc() 方法将此函数添加到 numpy 库。
使用 numpy 使用此函数。

frompyfunc() 方法

此函数允许创建任意Python函数作为 Numpy ufunc（通用函数）。此方法采用以下参数：

Parameters:

function – the name of the function that you create.
inputs – the number of input arguments (arrays) that function takes.
outputs – the number of output (arrays) that function produces.

编程需要懂一点英语

注意：有关更多信息，请参阅Python中的 numpy.frompyfunc() 。

例子：

创建一个名为 fxn 的函数，它接受一个值并返回一个值。
输入是一个一个的数组元素。
输出是使用某些逻辑修改的数组元素。

Python3

# using numpy
import numpy as np
 
# creating own function
def fxn(val):
  return (val % 2)
 
# adding into numpy
mod_2 = np.frompyfunc(fxn, 1, 1)
 
# creating numpy array
arr = np.arange(1, 11)
print("arr     :", *arr)
 
# using function over numpy array
mod_arr = mod_2(arr)
print("mod_arr :", *mod_arr)

输出：

arr     : 1 2 3 4 5 6 7 8 9 10
mod_arr : 1 0 1 0 1 0 1 0 1 0