python - numpy中数组之间的映射

Question

我在一个数组中有一些数据，我想映射到另一个数组，给定一个对应数组:

• originaldata 是一个 numpy 二维数组，
• targetdata 是另一个 numpy 二维数组，
• mapping 是一个在位置之间映射的数组，所以 mapping[x,y] 给了我一对 targetdata[x 的数据所在的坐标,y] 来自 originaldata.

到目前为止，我做了这样的事情:

for (x,y) in ALLTHEPOINTS:
    targetdata[x,y]=originaldata[mapping[x,y][0],mapping[x,y][1]]

...我怀疑这是非常低效的。

有什么方法可以对其进行矢量化吗？或者是否有解决此类操作的任何 numpy 函数？

Answer 1

这就是花式索引的作用:

targetdata = originaldata[mapping[..., 0], mapping[..., 1]]

举个简单的例子:

>>> original_data = np.arange(6).reshape(2, 3)
>>> original_data
array([[0, 1, 2],
       [3, 4, 5]])
>>> mapping = np.array([[[1,0], [1, 1], [1, 2]],   # swap rows and reverse
...                     [[0, 2], [0, 1], [0, 0]]]) # the second one
>>> original_data[mapping[..., 0], mapping[..., 1]]
array([[3, 4, 5],
       [2, 1, 0]])