如何在python中左移一个bitarray

时间:2013-12-18 18:36:02

标签: python des

我有一个名为c0的位数组,包含28位

bitarray('1111011111111111111111110001')

我怎么能多次移位这个位数组,这意味着一个左移,两个左移,等等? 现在一个左转很好!

注意:这与bitarray包相关。

2 个答案:

答案 0 :(得分:4)

你可以使用切片:

def leftshift(ba, count):
    return ba[count:] + (bitarray('0') * count)

def rightshift(ba, count):
    return (bitarray('0') * count) + ba[:-count]

这些保持输入的位宽,一端丢弃位,另一端用0填充。

您可以创建自己的bitarray类型的子类:

class mybitarray(bitarray):
    def __lshift__(self, count):
        return self[count:] + type(self)('0') * count
    def __rshift__(self, count):
        return type(self)('0') * count + self[:-count]
    def __repr__(self):
        return "{}('{}')".format(type(self).__name__, self.to01())

演示:

>>> c0 = mybitarray(c0)
>>> c0
mybitarray('1111011111111111111111110001')
>>> c0 << 4
mybitarray('0111111111111111111100010000')
>>> c0 >> 4
mybitarray('0000111101111111111111111111')

您也可以将作者戳到support these operations natively

答案 1 :(得分:1)

<<>>是移位运算符(bitarray类应该覆盖__ilshift____irshift__方法。