Python numpy：通过给定的索引列表重新排序 numpy 数组

Question

我有索引数组：

test_idxs = np.array([4, 2, 7, 5])

我还有值数组（更长）：

test_vals = np.array([13, 19, 31, 6, 21, 45, 98, 131, 11])

所以我想获取具有索引数组长度的数组，但按照索引数组的顺序获取值数组中的值（相当混乱），所以我想得到这样的东西：

array([21, 31, 131, 45])

我知道如何循环执行此操作，但我不知道如何使用 numpy 工具实现此目的。

Answer 1

这实际上对于 numpy 来说非常简单，只需用

test_vals

索引你的

test_idx

数组即可：

out = test_vals[test_idxs]

输出：

array([ 21,  31, 131,  45])