Python NumPy take方法用法及代码示例

例子

访问矩阵中的单个值

x = [[4,5,6], [7,8,9]]
np.take(x, 1)



5

请记住，如果您为第二个参数提供 number 类型，则 Numpy 会将您的输入数组视为展平数组，即 x=[4,5,6,7,8,9] 。在此示例中，我们采用该展平数组的第一个索引，因此返回值 5。

访问矩阵的单行

要访问矩阵的行，请提供参数 axis=0 。

x = [[4,5,6], [7,8,9]]
np.take(x, 1, axis=0)



array([7, 8, 9])

为了澄清这段代码的作用，我们只是获取第二行(即行也遵循索引顺序，所以这就是我们将 1 设置为第二个参数的原因)：

访问矩阵的单列

要访问矩阵的列，请提供参数 axis=1 。

x = [[4,5,6], [7,8,9]]
np.take(x, 1, axis=1)



array([5, 8])

为了澄清这段代码的作用，我们只是获取第二列(即列也遵循索引顺序，所以这就是我们将 1 设置为第二个参数的原因)：

访问矩阵的多行

要访问矩阵的多列，请提供参数 axis=0 ，并为第二个参数提供类型 array-like ：

x = [[4,5,6], [7,8,9]]
np.take(x, [0,1], axis=0)



array([[4, 5, 6],
       [7, 8, 9]])

我们正在访问第一行和第二行，如下所示：

访问矩阵的多列

要访问矩阵的多列，请提供参数 axis=1 ，并为第二个参数提供类型 array-like ：

x = [[4,5,6], [7,8,9]]
np.take(x, [0,2], axis=1)



array([[4, 6],
       [7, 9]])

我们正在访问第一列和第三列，如下所示：