首页 > 解决方案 > 将映射数组应用于numpy数组中的所有元素的快速方法?

问题描述

现在,我的代码基本上看起来像:

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

其中output和都是 numpy 数组(映射大小为 256,都是 uint8 类型)inputmap

这有效,但速度很慢。像这样的循环应该在 C 中。这就是 numpy 的用途。

是否有一个 numpy 函数(或 cv2 函数,我已经在导入它)可以做到这一点?

标签: pythonperformancenumpy

解决方案


怎么样?

output = map[input]

推荐阅读