Я хочу превратить байтовую строку, например b'\xed\x07b\x87S.\x866^\x84\x1e\x92\xbf\xc5\r\x8c'
, в пустой массив из 1 и 0 (т.е. двоичное значение этой байтовой строки в виде массива двоичных значений).
Как мне это сделать?
Я пытался использовать np.fromstring
и np.frombuffer
, но ни то, ни другое не помогло.
Используйте numpy.unpackbits. Согласно документам:
Распаковывает элементы массива uint8 в выходной массив с двоичным значением.
import numpy as np
b = b'\xed\x07b\x87S.\x866^\x84\x1e\x92\xbf\xc5\r\x8c'
bits_array = np.unpackbits(np.frombuffer(b, dtype=np.uint8))
print(bits_array)
выходы
[1 1 1 0 1 1 0 1 0 0 0 0 0 1 1 1 0 1 1 0 0 0 1 0 1 0 0 0 0 1 1 1 0 1 0 1 0
0 1 1 0 0 1 0 1 1 1 0 1 0 0 0 0 1 1 0 0 0 1 1 0 1 1 0 0 1 0 1 1 1 1 0 1 0
0 0 0 1 0 0 0 0 0 1 1 1 1 0 1 0 0 1 0 0 1 0 1 0 1 1 1 1 1 1 1 1 0 0 0 1 0
1 0 0 0 0 1 1 0 1 1 0 0 0 1 1 0 0]
Я не верю, что numpy дает вам возможность сделать это напрямую.
Вы можете сделать это в несколько шагов:
x = np.frombuffer(data, dtype=np.ubyte)
y = np.array([1, 2, 4, 8, 16, 32, 64, 128])
z = np.sign(x[:,None] & y[None,:])
Теперь это дает вам массив 16x8 из 0 и 1 с вашими данными. Вы можете изменить его размер, если вам нужны плоские данные.
z.resize(len(data) * 8)
Кроме того, вы можете использовать .reshape(-1)
, .flat
или .flatten()
вместо изменения размера с явным вычислением длины.
@Brian: было неясно, в каком порядке OP хотел получить выходные байты, поскольку образец вывода не был предоставлен. Я видел интерпретацию байтов как с прямым порядком байтов, так и с прямым порядком байтов.
@Brian Следите за тем, чтобы np.ndarray.flatten()
создавал копию ввода, а не представление, которое вернут np.ndarray.reshape(-1)
или np.ndarray.ravel()
. np.ndarray.flat
возвращает объект, похожий на итератор, поэтому он сильно отличается от других.
Я думаю, что
y
следует перевернуть. В противном случае вы получите[0, 1, 0, 0, 1, 0, 0, 0]
вместо\x12
вместо[0, 0, 0, 1, 0, 0, 1, 0]
.