当前位置：极客教程 > Numpy > Numpy 数组操作 > Python Numpy numpy.choice()

Python Numpy numpy.choice()

在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)