Изменение формы массива NumpyPython

Программы на Python
Anonymous
Изменение формы массива Numpy

Сообщение Anonymous »

У меня есть массив Numpy immatrix, который имеет immatrix.shape (4535,), и каждый элемент в массиве immatrix представляет собой другой массив imArray формы (12288,) . Как изменить форму массива имматрица до (4535,3,64,64)

Код: Выделить всё

    from PIL import Image
import PIL.Image as Pim

img_rows, img_cols = 64, 64
# number of channels
img_channels = 3

ret = []
for im2 in imlist_lettering:
img = Image.open(dirPath_of_non_spam + '/' + im2);
img = img.resize(( img_rows, img_cols), Pim.BILINEAR)
imArray = array(img, dtype=float).flatten();
if img_channels * img_cols *  img_rows == imArray.shape[0]:
ret.append(imArray)

immatrix = array(ret)
Здесь форма immatix — (4535), а форма imArray — (12288). Я хочу изменить форму immatix на (4535,3,64,64). Как это сделать??

Вернуться в «Python»