Как я понимаю, вам нужен список предопределенных цветов:
import numpy as np
import matplotlib.pyplot as plt
import random
colors = [
(1.0, 0.0, 0.0),
(0.0, 1.0, 0.0),
(0.0, 0.0, 1.0),
(1.0, 1.0, 0.0),
]
img = np.zeros(shape=(28, 28, 3))
# print(img)
for i in range(28):
for j in range(28):
img[i, j] = random.choice(colors)
plt.imshow(img)