select(condlist, choicelist, default=0)¶
Returns an array drawn from elements in choicelist, depending on conditions.
condlist (Union[int, float, bool, list, tuple, Tensor]) – The list of conditions which determine from which array in choicelist the output elements are taken. When multiple conditions are satisfied, the first one encountered in condlist is used.
default (scalar, optional) – The element inserted in output when all conditions evaluate to False.
Tensor, the output at position m is the m-th element of the array in choicelist where the m-th element of the corresponding array in condlist is True.
ValueError – if
len(condlist) != len(choicelist).
- Supported Platforms:
>>> import mindspore.numpy as np >>> condlist = [[True, True, True, False, False], [False, False, True, False, True]] >>> choicelist = [[0, 1, 2, 3, 4], [0, 1, 4, 9, 16]] >>> output = np.select(condlist, choicelist) >>> print(output) [ 0 1 2 0 16]