Python 如何用另一个数组中的值替换包含索引的数组?
我有一个数组b,包含数组a的索引。我想在数组b中插入另一个具有相同索引的数组c的值Python 如何用另一个数组中的值替换包含索引的数组?,python,python-2.7,numpy,Python,Python 2.7,Numpy,我有一个数组b,包含数组a的索引。我想在数组b中插入另一个具有相同索引的数组c的值 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] ##这将创建一个索引为2个最小值a1的数组 a [[0 3] [3 4] [1 0]] b=np.array([[1],[2],[3],[4],[5],[6]]) 现在我想用b中的1替换a中的值0;
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]
##这将创建一个索引为2个最小值a1的数组
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
相同的尺寸,可以对其进行重塑:
result = result.reshape(a.shape)
[[1, 4]
[4, 5]
[2, 1]]
你能再澄清一点吗?您提到了一个数组
c
,但没有在示例中显示它。你能在你的例子中展示一下你希望操作的输出是什么吗?太棒了。。。多谢各位