Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Faster way to loop through every pixel of an image in Python?

I need to loop through each pixel of a 2560x2160 2D numpy array (image). A simplified version of my problem is as follows:

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

This is taking an obnoxious ~30 seconds to complete on my computer. ( Core i7, 8GB ram ) Is there a faster way to perform this loop with an interior 'if' statement? I am only interested in pixels above a certain limit, but I do need their (x,y) indices and value.

like image 944
dinkelk Avatar asked Oct 22 '12 01:10

dinkelk


1 Answers

Use a boolean matrix:

x, y = (image > limit).nonzero()
vals = image[x, y]
like image 100
nneonneo Avatar answered Nov 15 '22 14:11

nneonneo