web-dev-qa-db-ja.com

Pythonでビットリストを整数に

私はPythonにそのようなリストがあります：[1,0,0,0,0,0,0,0]。 0b10000000と入力したように整数に変換できますか（128に変換できます）？ [1,1,0,0,0,0,0,0,1,0,0,0,0,0,0,0]のようなシーケンスを整数に変換する必要もあります（ここでは0b1100000010000000、つまり259を返します）。リストの長さは、必要な場合は常に8の倍数です。

pythonarrayslistbit-manipulation
21
ghostmansd

あなたはビットシフトを使うことができます：

out = 0
for bit in bitlist:
    out = (out << 1) | bit

これは、A。R. S.によって提案された「intキャスト」メソッド、またはSteven Rumbalskiによって提案されたルックアップを使用した修正キャストに勝るものです。

>>> def intcaststr(bitlist):
...     return int("".join(str(i) for i in bitlist), 2)
... 
>>> def intcastlookup(bitlist):
...     return int(''.join('01'[i] for i in bitlist), 2)
... 
>>> def shifting(bitlist):
...     out = 0
...     for bit in bitlist:
...         out = (out << 1) | bit
...     return out
... 
>>> timeit.timeit('convert([1,0,0,0,0,0,0,0])', 'from __main__ import intcaststr as convert', number=100000)
0.5659139156341553
>>> timeit.timeit('convert([1,0,0,0,0,0,0,0])', 'from __main__ import intcastlookup as convert', number=100000)
0.4642159938812256
>>> timeit.timeit('convert([1,0,0,0,0,0,0,0])', 'from __main__ import shifting as convert', number=100000)
0.1406559944152832
44
Martijn Pieters

...または bitstring モジュールを使用する

>>> from bitstring import BitArray
>>> bitlist=[1,0,0,0,0,0,0,0]
>>> b = BitArray(bitlist)
>>> b.uint
128
13
Fredrik Pihl

もちろん、Martijn Pietersのソリューションよりもわずかに優れた方法を見つけましたが、彼のソリューションはよりきれいです。私は実際に結果に少し驚いていますが、とにかく...

import timeit

bit_list = [1,1,0,0,0,0,0,0,1,0,0,0,0,0,0,0]

def mult_and_add(bit_list):
    output = 0
    for bit in bit_list:
        output = output * 2 + bit
    return output

def shifting(bitlist):
     out = 0
     for bit in bitlist:
         out = (out << 1) | bit
     return out

n = 1000000

t1 = timeit.timeit('convert(bit_list)', 'from __main__ import mult_and_add as convert, bit_list', number=n)
print "mult and add method time is : {} ".format(t1)
t2 = timeit.timeit('convert(bit_list)', 'from __main__ import shifting as convert, bit_list', number=n)
print "shifting method time is : {} ".format(t2)

結果：

mult and add method time is : 1.69138722958 
shifting method time is : 1.94066818592
6
Akavall

このワンライナーを試してください：

int("".join(str(i) for i in my_list), 2)

速度/効率に関心がある場合は、Martijn Pietersのソリューションをご覧ください。

4
arshajii

これはどう：

out = sum([b<<i for i, b in enumerate(my_list)])

または逆の順序で：

out = sum([b<<i for i, b in enumerate(my_list[::-1])])
0
Weidong He