Python numpy piecewise用法及代碼示例

本文簡要介紹 python 語言中 numpy.piecewise 的用法。

用法:

numpy.piecewise(x, condlist, funclist, *args, **kw)

評估 piecewise-defined 函數。

給定一組條件和相應的函數，在輸入數據的條件為真時評估每個函數。

參數：

x： ndarray 或標量

輸入域。

condlist： 布爾數組或布爾標量列表

每個布爾數組對應 funclist 中的一個函數。當 condlist[i] 為 True 時，funclist[i](x) 被用作輸出值。

condlist 中的每個布爾數組選擇 x 的一部分，因此應該與 x 具有相同的形狀。

的長度條件清單必須對應於函數列表.如果給出一個額外的函數，即如果len(funclist) == len(condlist) + 1，那麽這個額外的函數就是默認值，在所有條件都為假的地方使用。

funclist： 可調用對象、f(x,*args,**kw) 或標量列表

每個函數都經過評估x隻要其對應條件為真。它應該將一維數組作為輸入，並給出一維數組或標量值作為輸出。如果提供標量而不是可調用對象，則提供一個常量函數 (lambda x: scalar) 假設。

args： 元組，可選

給 piecewise 的任何其他參數都會在執行時傳遞給函數，即，如果調用 piecewise(..., ..., 1, 'a') ，則每個函數都稱為 f(x, 1, 'a') 。

kw： 字典，可選

調用 piecewise 時使用的關鍵字參數在執行時傳遞給函數，即，如果調用 piecewise(..., ..., alpha=1) ，則每個函數都稱為 f(x, alpha=1) 。

返回：

out： ndarray

輸出的形狀和類型與 x 相同，可以通過在 x 的適當部分調用 funclist 中的函數來找到，如 condlist 中的布爾數組所定義。任何條件未涵蓋的部分的默認值為 0。

注意：

這類似於選擇或選擇，除了函數是在滿足 condlist 中相應條件的 x 元素上評估的。

結果是：

|--
      |funclist[0](x[condlist[0]])
out = |funclist[1](x[condlist[1]])
      |...
      |funclist[n2](x[condlist[n2]])
      |--

例子：

定義 sigma 函數，對於 x < 0 為 -1，對於 x >= 0 為 +1。

>>> x = np.linspace(-2.5, 2.5, 6)
>>> np.piecewise(x, [x < 0, x >= 0], [-1, 1])
array([-1., -1., -1.,  1.,  1.,  1.])

定義絕對值，對於 x <0 是 -x ，對於 x >= 0 是 x 。

>>> np.piecewise(x, [x < 0, x >= 0], [lambda x: -x, lambda x: x])
array([2.5,  1.5,  0.5,  0.5,  1.5,  2.5])

將相同的函數應用於標量值。

>>> y = -2
>>> np.piecewise(y, [y < 0, y >= 0], [lambda x: -x, lambda x: x])
array(2)