我需要使用索引矩阵从 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
函数,但结果不正确。
答案 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])