Мне нужно отобразить двоичное изображение с указанным c цветом.
На данный момент я создаю изображение RGB с нужным цветом:
from matplotlib import pyplot as plt
import numpy as np
image = np.random.normal(0, 10, (20, 20)) > 0 # generate a (20,20) image with random value
binary = image > 0 # create a binary image
binary_rgb = np.zeros(binary.shape + (3,)) #prepare the RGB image
rgb_color = [1, 0, 0] # we choose red for the binary image
binary_rgb[binary,:] = rgb_color
plt.figure()
plt.imshow(binary_rgb)
Есть ли функция в matplotlib что позволило бы сделать это легко?
Спасибо!