python - 如何使用 numpy.where 更改图像的所有像素？

Question

我有一个形状为 (300,300,3) 的图像，由这些像素组成 [255, 194, 7],[224, 255, 8],[230, 230, 230],[11, 102, 255]。我想将此像素 [230, 230, 230] 更改为 [255,255,255]。将其他像素保留到[0,0,0]。所以我应用 numpy where 函数来切换像素。下面是代码:

import numpy

im = numpy.array([[[255, 194, 7],[224, 255, 8],[230, 230, 230],[11, 102, 255]]])

im[np.where((im == [230, 230, 230]).all(axis = 2))] = [255,255,255]
im[np.where((im != [255,255,255]).all(axis = 2))] = [0,0,0]

第一个代码工作正常，但所有包含 255 的像素(例如 [11, 102, 255])在第二行中根本不会翻转。并且图像保持不变。谁能告诉我我做错了什么？

Answer 1

import numpy as np
im = np.array([[[255, 194, 7],[224, 255, 8],[230, 230, 230],[11, 102, 255]]])

像这样吗？
制作一个掩码并用它来更改值。

>>> mask = im == 230
>>> im[mask] = 255
>>> im[np.logical_not(mask)] = 0

>>> im
=> array([[[  0,   0,   0],
        [  0,   0,   0],
        [255, 255, 255],
        [  0,   0,   0]]])

或者使用 numpy.where

>>> np.where(im==230, 255, 0)

=> array([[[  0,   0,   0],
        [  0,   0,   0],
        [255, 255, 255],
        [  0,   0,   0]]])