Numpy入门[16]——choose函数实现条件筛选Numpy入门[16]——choose函数实现条件筛选

134 阅读 0 评论 89 点赞

我是靠谱客的博主热情小蝴蝶，这篇文章主要介绍Numpy入门[16]——choose函数实现条件筛选Numpy入门[16]——choose函数实现条件筛选，现在分享给大家，希望可以做个参考。

Numpy入门[16]——choose函数实现条件筛选

参考：

https://ailearning.apachecn.org/
numpy高级函数操作之——select、choose

使用Jupyter进行练习

借助numpy.choose()方法，我们可以通过将包含要选择的行号索引的参数作为数组传递，从多维数组中选择元素。

对于数组，有时候需要进行类似 switch 和 case 进行条件选择，此时使用 choose 函数十分方便：

复制代码

1
2
def choose(a, choices, out=None, mode='raise')

参数 a ：它必须是一个 int 型的数组，并且 a 中的元素，必须是0~n-1之间的数，这里的n表示的就是数组choices数组最外层的维度数。

choices：表示的是要操作的数组，要注意的是choices的数组的维度是一定要和a进行匹配的，如果匹配不了，会出现错误。

参数out：接收运算结果的数组，它的维度一定要和 a 是一样的，是可选参数。

参数mode：

默认的是raise，表示的是a数组中的元素不能超过 n
clip：将 a 中的元素如果小于0，则将其变为0，如果大于n-1，则变为n-1
wrap：将a中的值 value变为value mod n，即值除以n的余数。

复制代码

1
2
3
4
5
6
7
8
9
10
import numpy as np
a = np.array([[2,0,1],
              [2,1,0],
              [1,2,2]])
choice = np.array([])

np.choose(a, [[0, 1, 2], 
              [10, 11, 12],
              [20, 21, 22]])

复制代码

1
2
3
4
array([[20,  1, 12],
       [20, 11,  2],
       [10, 21, 22]])

在上面的例子中，a中第一行第一列元素映射为choice中第三行（2）第一列元素，第一行第二列元素映射为choice中第一行（0）第二列元素，映射的是行下标。

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

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
i0 = np.array([[0,1,2],
               [3,4,5],
               [6,7,8]])
i2 = np.array([[20,21,22],
               [23,24,25],
               [26,27,28]])
a = np.array([[1,0,1],
              [2,1,0],
              [1,2,2]])

np.choose(a, [i0, 10, i2])

复制代码

1
2
3
4
array([[10,  1, 10],
       [23, 10,  5],
       [10, 27, 28]])

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

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

复制代码

1
2
3
4
5
6
a = np.array([[ 0, 1, 2], 
              [10,11,12], 
              [20,21,22]])

a < 10

复制代码

1
2
3
4
array([[ True,  True,  True],
       [False, False, False],
       [False, False, False]])

复制代码

1
2
3
# False(0)对应的是a中相应的元素，True(1)对应的是10
np.choose(a < 10, (a, 10))

复制代码

1
2
3
4
array([[10, 10, 10],
       [10, 11, 12],
       [20, 21, 22]])

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

复制代码

1
2
3
4
5
6
7
8
9
10
a = np.array([[ 0, 1, 2], 
              [10,11,12], 
              [20,21,22]])

lt = a < 10
gt = a > 15

choice = lt + 2 * gt
choice

复制代码

1
2
3
4
array([[1, 1, 1],
       [0, 0, 0],
       [2, 2, 2]])

复制代码

1
2
np.choose(choice, (a, 10, 15))

复制代码

1
2
3
4
array([[10, 10, 10],
       [10, 11, 12],
       [15, 15, 15]])

最后

以上就是热情小蝴蝶最近收集整理的关于Numpy入门[16]——choose函数实现条件筛选Numpy入门[16]——choose函数实现条件筛选的全部内容，更多相关Numpy入门[16]——choose函数实现条件筛选Numpy入门[16]——choose函数实现条件筛选内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。

本文分类：# python机器学习
浏览次数：134 次浏览
发布日期：2023-11-01 09:36:22
本文链接：https://www.kaopuke.com/article/k-p-k_13_u_23_o_18_f2_13__23__2_z.html

Numpy入门[16]——choose函数实现条件筛选Numpy入门[16]——choose函数实现条件筛选

Numpy入门[16]——choose函数实现条件筛选

最后

评论列表共有 0 条评论

发表评论取消回复

Numpy入门[16]——choose函数实现条件筛选Numpy入门[16]——choose函数实现条件筛选

Numpy入门[16]——choose函数实现条件筛选

最后

相关文章

评论列表共有 0 条评论

发表评论 取消回复

微信扫一扫：分享

发表评论取消回复