Как изменить изображение с (n, m, 4) на (n, m, 3) в Python?

Question

У меня есть значение изображения, подобное следующему:

A = [[[  7,  23,   0, 255],
         [ 14,  25,   1, 255],
         [ 23,  26,   1, 255],
         [ 49,  41,  33, 255],
         [ 43,  37,  30, 255],
         [ 42,  33,  25, 255]],

        [[ 59,  69,  59, 255],
         [ 77,  83,  72, 255],
         [ 86,  91,  84, 255],
         [ 16,  16,   8, 255],
         [  9,   7,   2, 255],
         [  1,   0,   0, 255]]]

print(np.array(A).shape) ## (3,6,4)

Я хочу изменить его, например (3,6,3), чтобы пропустить 255 для матрицы.

Какя могу сделать?

Answer 1

Вам нужно нарезать, а не изменять форму, которая сохраняет все элементы, но меняет форму массива;Для вашего случая просто перетащите последний элемент вдоль третьей оси:

np.array(A)[:,:,:-1].shape
# (2, 6, 3)