ecarlson at eng.ua.edu
Sat Nov 19 10:55:52 EST 2011
I'm not sure if I am addressing your question on vectorizing directly,
but consider the following code, which does (maybe?) what your asking.
from numpy import reshape,ones, zeros, arange, array
B=zeros(A.shape) #initialize array
filt=ones([3,3])/9.0 #the weighting matrix
#initialize variance matrix for interior elements
for i in arange(-1,2):
for j in arange(-1,2):
B_var/=9.0 #variance of 8x8 interior elements
This may or may not be the best or fastest way to get your answer. In
particular, it may be just as fast to use the same loop structure for B
as for B_var since convolve2D is not known as a speed demon. On the
whole, though, I usually like to use scipy/numpy functions when available.
If you need to do this calculation a lot, it would be pretty
straightforward to parallelize with c or cuda. You'd just need to take
care with minimizing memory accesses (by doing things like calculating
average and variance after doing one fetch of the 9 elements).
More information about the NumPy-Discussion