python - 在 Python 中循环遍历图像的每个像素的更快方法？

Question

我需要遍历 2560x2160 二维 numpy 数组(图像)的每个像素。我的问题的简化版本如下:

import time
import numpy as np

t = time.clock()
limit = 9000
for (x,y), pixel in np.ndenumerate(image):
    if( pixel > limit )
        pass
tt = time.clock()
print tt-t

这在我的电脑上花费了约 30 秒的时间，令人讨厌。 (酷睿 i7、8GB 内存)有没有更快的方法来使用内部“if”语句执行此循环？我只对超过某个限制的像素感兴趣，但我确实需要它们的 (x,y) 索引和值。

Answer 1

使用 bool 矩阵:

x, y = (image > limit).nonzero()
vals = image[x, y]