我有一个数组 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 将这些索引作为值处理
请帮忙
最佳答案
如果我理解正确,您可以使用 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]]
关于python - 如何用另一个数组中的值替换包含索引的数组?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/24345735/