numpy图像 - 根据其他图像替换颜色

Question

我试图创建显示2张图片之间差异的图片。我有几个函数可以比较2种颜色（3个数字的数组）并返回它们的差异0.0 - 1.0。

compare_colors(color1, color2)
    # do some calculation
    return difference

现在我想用常量颜色替换第一个图像中的像素（让我们说[255, 0, 255]），其中色差高于使用选定tolerance定义的compare_colors功能

different = some selector where compare_colors(image1, image2) > tolerance
original_image[different] = [255, 0, 255]

但是，我不知道如何使用compare_colors功能创建选择器。我还需要访问compare_colors函数中的各个颜色通道。