如何在python中创建带有负索引/输入的查找表？

Question

我有一组输入矩阵A，其中可能包含负元素。我还有一组从int到int的映射，我希望有效地应用于A。 示例：

import numpy as np

ind = np.array([-9, -8, -7, -6, -5, -4, -3, -2, -1])
out = np.array([ 1,  2,  3,  4,  5,  6,  7,  8,  9])

# i-th element of ind should return i-th element of out

a = np.array([[-1, -2, -3], [-4, -5, -6], [-7, -8, -9]])
# print(a)
# array([[-1, -2, -3],
#        [-4, -5, -6],
#        [-7, -8, -9]])

# i want output as 
# array([[ 9,  8,  7],
#        [ 6,  5,  4],
#        [ 3,  2,  1]])

对不起，如果我说不清楚的话。 不需要有一个函数来控制从ind到out的转换。你知道吗

我现在唯一能想到的就是做一个dict并迭代输入矩阵的所有元素。但那会很慢。如何有效地做到这一点？你知道吗

Answer 1

我们可以用^{}-

In [43]: out[np.searchsorted(ind,a)]
Out[43]: 
array([[9, 8, 7],
       [6, 5, 4],
       [3, 2, 1]])

对于ind不一定排序的一般情况，我们需要使用sorterarg-

In [44]: sidx = ind.argsort()

In [45]: out[sidx[np.searchsorted(ind,a,sorter=sidx)]]
Out[45]: 
array([[9, 8, 7],
       [6, 5, 4],
       [3, 2, 1]])