python - 如何用另一个数组中的值替换包含索引的数组？

Question

我有一个数组 b，其中包含数组 a 的索引。我想将另一个数组 c 的值插入到具有相同索引的数组 b 中。

import numpy as np
a1=np.array([[1, 3, 5, 2, 3],[7, 6, 5, 2, 4],[2, 0, 5, 6, 4]])
a=a1.argsort()[:,:2]

## 这将创建一个数组，其索引为 a1 的 2 个最小值

  a
[[0 3]
 [3 4]
 [1 0]]
b=np.array([[1],[2],[3],[4],[5],[6]])

现在我想用 b 中的 1 替换 a 中的值 0 ； 3 与 4 依此类推

我尝试使用:

[a[index]]=b[index]

但这显然不是正确的方式，因为数组 a 将这些索引作为值处理

请帮忙

Answer 1

如果我理解正确，您可以使用 a 的扁平版本来索引 b:

result = b.ravel()[a.ravel()]

[1, 4, 4, 5, 2, 1]

如果您需要它与 a 具有相同的尺寸，您可以 reshape 它:

result = result.reshape(a.shape)

[[1, 4]
 [4, 5]
 [2, 1]]