numpy.ndarray.choose

ndarray.choose(choices, out=None, mode='raise')

使用索引数组从一组选择中构造新的数组。

有关完整文档，请参阅numpy.choose。

也可以看看

numpy.choose

等效函数