#python #arrays #image #numpy
#питон #массивы #изображение #numpy #python
Вопрос:
У меня есть массив, который я могу обработать следующим образом:
ba = bytearray(fh.read())[32:]
size = int(math.sqrt(len(ba)))
Я могу сказать, должен ли пиксель быть черным или белым, учитывая
iswhite = (ba[i]amp;1)==1
Как я могу быстро преобразовать свой массив 1D byte в массив 2D numpy с длиной строки size
и белыми пикселями для (ba[i]amp;1)==1
и черными для других? Я создаю массив следующим образом:
im_m = np.zeros((size,size,3),dtype="uint8)
Ответ №1:
import numpy as np
# fh containts the file handle
# go to position 32 where the image data starts
fh.seek(32)
# read the binary data into unsigned 8-bit array
ba = np.fromfile(fh, dtype='uint8')
# calculate the side length of the square array and reshape ba accordingly
side = int(np.sqrt(len(ba)))
ba = ba.reshape((side,side))
# toss everything else apart from the last bit of each pixel
ba amp;= 1
# make a 3-deep array with 255,255,255 or 0,0,0
img = np.dstack([255*ba]*3)
# or
img = ba[:,:,None] * np.array([255,255,255], dtype='uint8')
Есть несколько способов выполнить последний шаг. Просто будьте осторожны, вы получаете тот же тип данных ( uint8
), если вам это нужно.
Комментарии:
1. Большое спасибо, именно то, что мне было нужно
2. Ну, у меня небольшая проблема, вот полный код: pastebin.com/qX69JxpZ Я пытаюсь экспортировать в JPG, но получаю сообщение об ошибке «Максимальный поддерживаемый размер изображения составляет 65500 пикселей»
3. Структура, по-видимому, представляет собой один внешний массив с массивом внутри него массивов пикселей (3 числа)
4. @Кристиан Стюарт. У меня была опечатка в моем коде в строке с reshape. Я исправил это. Дело в том, что
a.reshape(x,x)
возвращает измененную версиюa
(не изменяет ее), тогда какa.resize((x,x))
илиa = a.reshape((x,x))
действительно изменяет форму A. Извините за ошибку!