choose 函数实现条件筛选

对于数组,我们有时候需要进行类似 switchcase 进行条件选择,此时使用 choose 函数十分方便:

numpy的choose 函数实现条件筛选 

在上面的例子中,choose0,1,2 对应的值映射为了 10, 11, 12,这里的 0,1,2 表示对应的下标。

事实上, choose 不仅仅能接受下标参数,还可以接受下标所在的位置:

numpy的choose 函数实现条件筛选

这里,control 传入第一个 1 对应的是 10,传入的第一个 0 对应于 i0 相应位置的值即 1,剩下的以此类推。

下面的例子将数组中所有小于 10 的值变成了 10

numpy的choose 函数实现条件筛选 

下面的例子将数组中所有小于 10 的值变成了 10,大于 15 的值变成了 15。

numpy的choose 函数实现条件筛选 


 

 

 

相关文章: