从带有索引矩阵的 numpy 矩阵中获取元素

时间:2021-02-23 09:07:24

标签: python numpy matrix

我需要使用索引矩阵从 numpy 矩阵中获取一些元素。这是我想象的一个例子:

import numpy as np
    
matrix = np.random.rand(100,8)
indices = np.array([[10,0],[17,1],[24,2],[50,3]])

从这里,我想找到一种方法,用一行从矩阵值中获取索引矩阵中的元素,其中该结果矩阵的第一个元素将是元素矩阵[10,0] ,第二个矩阵[17,1],依此类推。

我尝试使用 np.take 函数,但结果不正确。

1 个答案:

答案 0 :(得分:0)

这有效:

import numpy as np

matrix = np.random.rand(100,8)
indices = np.array([[10,0],[17,1],[24,2],[50,3]])

print(matrix[indices])