当前位置：极客教程 > Numpy > Numpy 矩阵函数 > Python Numpy matrix.take()

Python Numpy matrix.take()

在Numpy matrix.take()方法的帮助下，我们可以通过传递参数作为该元素的索引值，从一个给定的矩阵中选择元素。它将返回一个一维的矩阵。记住，它一次只对一个轴起作用。

语法: matrix.take(index, axis)
返回： 返回所选索引的矩阵。

例子#1 :
在这个例子中，我们可以看到，通过选择一个索引，我们使用matrix.take()方法在矩阵中只得到一个值。

# import the important module in python
import numpy as np
              
# make matrix with numpy
gfg = np.matrix('[4, 1, 12, 3, 4, 6, 7]')
              
# applying matrix.take() method
geek = gfg.take(2)
    
print(geek)

输出:

[[12]]

例子#2 :

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