python - 将映射数组应用于 numpy 数组中的所有元素的快速方法？

Question

现在，我的代码基本上如下所示:

for x in range(img.shape[0]):
    for y in range(image.shape[1]):
        output[x,y] = map[ input[x,y] ]

其中output、input和map都是numpy数组(map的大小为256，都是uint8类型)。

这可行，但速度很慢。像这样的循环应该在 C 中。这就是 numpy 的用途。

是否有一个 numpy 函数(或 cv2 函数，无论如何我已经导入它)可以做到这一点？

Answer 1

怎么样？

output = map[input]