Как сгенерировать весь цвет в отдельный пиксель?

Question

Я могу сгенерировать все черные пиксели, сложив вместе 3 np.zeros ([1,1,1])

Но почему бы и нет?

import numpy as np
import cv2

m  = np.zeros([8,32,1])

k = 0 
for i in range(m.shape[0]):
    for j in range(m.shape[1]):
        for z in range(m.shape[2]):
            m[i,j,z] = k
            k+=1
a = np.stack(m,m,m)

cv2.imwrite('rgb.png',a)

Ошибка

axis = normalize_axis_index(axis, result_ndim)
TypeError: only size-1 arrays can be converted to Python scalars

Answer 1

ОК, поэтому я немного изменил ваш код, чтобы показать, как он работает:

m  = np.zeros([8,32])

k = 0 
for i in range(m.shape[0]):
    for j in range(m.shape[1]):
        m[i,j] = k
        k+=1
a = np.stack((m,m,m), axis=-1)

Вывод: