我正在尝试使用NumPy对循环迭代进行矢量化,但我正在努力实现所需的结果.我有一个像素值数组,所以3维,比如说(512,512,3)并且需要迭代每个x,y并使用第三维中的特定索引计算另一个值.标准循环中此代码的示例如下:
for i in xrange(width):
for j in xrange(height):
temp = math.sqrt((scalar1-array[j,i,1])**2+(scalar2-array[j,i,2])**2)
我目前正在做的是:
temp = np.sqrt((scalar1-array[:,:,1])**2+(scalar2-array[:,:,2])**2)
我从中获得的临时数组是所需的维度(x,y),但是一些值与循环实现不同.如何消除循环以在NumPy中有效地计算此示例？
提前致谢!
编辑:
这里的代码给出了temp和temp2的不同结果,显然temp2只是一个单元格的计算
temp = np.sqrt((cb_key-fg_cbcr_array[:,:,1])**2+(cr_key-fg_cbcr_array[:,:,2])**2)
temp2 = np.sqrt((cb_key-fg_cbcr_array[500,500,1])**2+(cr_key-fg_cbcr_array[500,500,2])**2)
print temp[500, 500]
print temp2
上面的输出是
12.039
94.069123521
标量肯定是初始化的,并且数组是使用图像生成的
fg = PIL.Image.open('fg.jpg')
fg_cbcr = fg.convert("YCbCr")
fg_cbcr_array = np.array(fg_cbcr)
EDIT2:
好的,我已经跟踪了我的阵列问题.不知道为什么然而它在使用np.random.random生成数组时有效,但在使用PIL从文件加载时却没有,如上所述.