Python Numpy numpy.choose()用法及代码示例

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

用法： numpy.choose()

返回： 返回元素选择的数组

范例1：
在此示例中，我们可以借助numpy.choose()方法，我们能够从多维数组中提取选择数组。

# import the important module in python 
import numpy as np 
             
# make a matrix with numpy 
gfg = [[1, 2, 3, 4], [3, 1, 5, 6]] 
             
# applying numpy.choose() method 
geeks = np.choose([1, 0, 1, 0], gfg) 
       
print(geeks)

输出：

array([3 2 5 4])

范例2：

# import the important module in python 
import numpy as np 
             
# make a matrix with numpy 
gfg = [[1, 2, 3], [4, 5, 6], [7, 8, 9]] 
             
# applying numpy.choose() method 
geeks = np.choose([2, 0, 1], gfg) 
       
print(geeks)

输出：

array([7 2 6])

相关用法

注：本文由纯净天空筛选整理自Jitender_1998大神的英文原创作品 Python | Numpy numpy.choose()。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。