这种NumPy列表理解的捷径

Question

有没有更简单，更快捷的方法？

maxr, maxc = im_out.shape[:2]

for col in range(maxc):
    for row in range(maxr):
        if im_gray[row,col,0] != 255 and im_gray[row,col,1] != 255 and im_gray[row,col,2] != 255:
            im_out[row, col] = im_gray[row, col]

Answer 1

遵循以下方法应该可以解决问题：

# I think it's axis 2, might have to play around there
mask = (im_gray != 255).all(axis=2)
im_out[mask] = im_gray[mask]

Answer 2

您可以使用遮罩：

mask = (im_gray[..., 0] != 255) & (im_gray[..., 1] != 255) & (im_gray[..., 2] != 255)

im_out[mask] = im_gray[mask]

上面的矢量化找到一个掩码，其中所有通道都满足条件。