Numpy矩阵获取另一个数组中的元素键集

Question

我有numpy矩阵（value_matrix）。

[[0.3,0.5,1,0.2,0.9,0.5,0.6,0.3,0.7,0.8]
 [0.9,0.6,0.1,0.4,0.2,0.7,0.6,0.9,0.5,0.4]]

我需要从每一行（在循环内）中获取列号（索引）在另一个numpy数组中的元素。

ex：key_array -> [2,0,1,3]

ex：value_matrix.item(0,key_array)一种检索方法。

我怎么能用numpy做到这一点？

Answer 1

据我了解，你需要这样的东西：

mat = np.matrix([[0.3,0.5,1,0.2,0.9,0.5,0.6,0.3,0.7,0.8],[0.9,0.6,0.1,0.4,0.2,0.7,0.6,0.9,0.5,0.4]]) 
key_array =  [2,0,1,3]
for i in range(len(mat)):
    print(mat[i,key_array])
# result
# [[ 1.   0.3  0.5  0.2]]
# [[ 0.1  0.9  0.6  0.4]]

然而，正如@Divakar所说，你的问题不够明确。