Python numpy select用法及代碼示例

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

用法:

numpy.select(condlist, choicelist, default=0)

根據條件返回從選擇列表中的元素中提取的數組。

參數：

condlist： bool ndarrays 列表

確定從選擇列表中的哪個數組獲取輸出元素的條件列表。當滿足多個條件時，使用 condlist 中遇到的第一個條件。

choicelist： ndarrays 列表

從中獲取輸出元素的數組列表。它必須與 condlist 的長度相同。

default： 標量，可選

當所有條件評估為 False 時插入到輸出中的元素。

返回：

output： ndarray

位置 m 的輸出是choicelist 中數組的m-th 元素，其中condlist 中相應數組的m-th 元素為True。

例子：

>>> x = np.arange(6)
>>> condlist = [x<3, x>3]
>>> choicelist = [x, x**2]
>>> np.select(condlist, choicelist, 42)
array([ 0,  1,  2, 42, 16, 25])

>>> condlist = [x<=4, x>3]
>>> choicelist = [x, x**2]
>>> np.select(condlist, choicelist, 55)
array([ 0,  1,  2,  3,  4, 25])