python 将图片中的颜色进行替换

from PIL import Image
import PIL.ImageOps    
import numpy as np
from skimage.io import imsave
import cv2


in_path  = 'y.png'
out_path = 'y1-trial.png'


Image = cv2.imread(in_path)
Image2 = np.array(Image, copy=True)
#白色
white_px = np.asarray([255, 255, 255])
#黑色
black_px = np.asarray([0  , 0  , 0  ])


(row, col, _) = Image.shape

for r in range(row):
    for c in range(col):
        px = Image[r][c]
        #检测白色，然后替换为黑色
        if all(px != white_px):
            Image2[r][c] = black_px
      
imsave(out_path, Image2)