建立您自己的 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 用作 numpy 的別名,則為 np.ufunc
)
示例
使用 if 語句檢查函式是否為 ufunc
import numpy as np
if type(np.add) == np.ufunc
print('add 是 ufunc')
else
print('add 不是 ufunc')
自己動手試一試 »