python - 将二进制 (0|1) numpy 转换为整数或二进制字符串？

Question

是否有将二进制 (0|1) numpy 数组转换为整数或二进制字符串的快捷方式？F.E.

b = np.array([0,0,0,0,0,1,0,1])   
  => b is 5

np.packbits(b)

有效，但仅适用于 8 位值。如果 numpy 有 9 个或更多元素，它会生成 2 个或更多 8 位值。另一种选择是返回一个字符串 0|1 ...

我目前做的是:

    ba = bitarray()
    ba.pack(b.astype(np.bool).tostring())
    #convert from bitarray 0|1 to integer
    result = int( ba.to01(), 2 )

这很丑!!!

Answer 1

一种方法是使用 dot-product使用 2-powered 范围数组 -

b.dot(2**np.arange(b.size)[::-1])

sample 运行-

In [95]: b = np.array([1,0,1,0,0,0,0,0,1,0,1])

In [96]: b.dot(2**np.arange(b.size)[::-1])
Out[96]: 1285

或者，我们可以使用按位左移运算符来创建范围数组，从而获得所需的输出，就像这样 -

b.dot(1 << np.arange(b.size)[::-1])

如果对时间感兴趣——

In [148]: b = np.random.randint(0,2,(50))

In [149]: %timeit b.dot(2**np.arange(b.size)[::-1])
100000 loops, best of 3: 13.1 µs per loop

In [150]: %timeit b.dot(1 << np.arange(b.size)[::-1])
100000 loops, best of 3: 7.92 µs per loop

---

逆向过程

要检索二进制数组，请使用 np.binary_repr连同np.fromstring -

In [96]: b = np.array([1,0,1,0,0,0,0,0,1,0,1])

In [97]: num = b.dot(2**np.arange(b.size)[::-1]) # integer

In [98]: np.fromstring(np.binary_repr(num), dtype='S1').astype(int)
Out[98]: array([1, 0, 1, 0, 0, 0, 0, 0, 1, 0, 1])