使用 Numpy 进行快速列表检查

Question

假设我有一张图像，我想检查它的每个像素以检查它是否符合我的需要。在 Python 中使用“for”需要很多时间，而我在 C++ 中做同样的事情要快得多。

我听说 Numpy 正在用于它，但我不确定如何

基本上：

for x in range(0, width):
    for y in range(0, height):
        if img[x][y][0] > value:
            ...do something...

使用这种简单的 Python 格式，一帧大约需要 5 秒，而 C++ 以毫秒为单位。

Answer 1

假设您有一个数组array = np.array([[1,2,3],[4,5,6]])，它是一个可以是图像的 2x3 矩阵。使用 numpy，您可以通过编写 array>threshold 来简单地检查一个值是否超过阈值，这将返回一个具有真假的 2x3 矩阵。您还可以编写array[array>threshold] = some new value，它将数组中超出阈值的所有值设置为特定值