[Numpy-discussion] two questions about `choose`

Alan G. Isaac alan.isaac at gmail.com
Sat Apr 17 13:55:40 EDT 2021

1. Is there a technical reason for `choose` not accept a `dtype` argument?

2. Separately, mypy is unhappy with my 2nd argument to `choose`:
Argument 2 to "choose" has incompatible type "Tuple[int, Sequence[float]]";
expected "Union[Union[int, float, complex, str, bytes, generic], Sequence[Union[int, float, complex, str, bytes, generic]], Sequence[Sequence[Any]],_SupportsArray]"
However, `choose` is happy to have e.g. `choices=(0,seq)` (and I hope it will remain so!).

a = a = (0,1,1,0,0,0,1,1)  #binary array
np.choose((0,range(8))     #array([0, 1, 2, 0, 0, 0, 6, 7])

Thanks, Alan Isaac

More information about the NumPy-Discussion mailing list