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

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

1.通常使用def关键字来定义该函数。
2.使用frompyfunc()方法将此函数加入numpy库。
3.用numpy来使用这个函数。

frompyfunc() 方法

这个函数允许创建一个任意的Python函数作为Numpy ufunc(通用函数)。该方法接受以下参数 :

参数:

• function – 你所创建的函数的名称。
• inputs – 该函数接受的输入参数（数组）的数量。
• outputs – 该函数产生的输出（数组）的数量。

注意：更多信息请参考Python中的numpy.frompyfunc()。

示例 :

• 创建名称为fxn的函数，取一个值，同时返回一个值。
• 输入是一个一个的数组元素。
• 输出是使用一些逻辑修改的数组元素。

# 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