Python Numpy matrix.choice()

在Numpy matrix.choose()方法的帮助下，我们可以通过传递一个包含待选行号索引的数组参数来选择矩阵中的元素。输出数组的大小与参数中传递的相同。

语法: matrix.choose()

返回 : 返回一个选择元素的数组

例子#1 :
在这个例子中，我们可以看到，在matrix.choose()方法的帮助下，我们能够从矩阵中提取一个选择数组。

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

输出:

[[3 2 5 4]]

例子#2 :

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

输出:

[[7 2 6]]