ufunc 创建函数

创建于 2024-12-03 / 35
字体: [默认] [大] [更大]

如何创建自己的 ufunc

要创建您自己的 ufunc,您必须定义一个函数,就像在 Python 中使用普通函数一样,然后使用 frompyfunc()方法将其添加到您的 NumPy ufunc 库中。

frompyfunc() 方法采用以下参数:

  1. function - 函数的名称。
  2. inputs - 输入参数(数组)的数量。
  3. 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:

import numpy as np

print(type(np.add)) 亲自试一试 »

如果不是 ufunc,它将返回另一种类型,例如用于连接两个或多个数组的内置 NumPy 函数:

实例

检查另一个函数的类型:concatenate():

import numpy as np

print(type(np.concatenate)) 亲自试一试 »

如果函数根本无法识别,则会返回错误:

实例

检查不存在的事物的类型。 这将产生错误:

import numpy as np

print(type(np.blahblah)) 亲自试一试 »

要在 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') 亲自试一试 »

0 人点赞过