Python轻量级的bit数组

2024-06-15

字数统计: 323字 | 阅读时长: 1分

bit数组（位数组）是一个保存大规模对象（通常以万计）状态的有力工具，本文采用内置的bytearray来实现一个轻量级的bit数组，当然你也可以选择使用已发布的bitarray（使用pip进行安装即可），该组件采用c语言编写，功能齐全、性能良好，值得推荐。

既然已经有了类似的工具，为啥还要重复造轮子？

为了使项目尽可能少的依赖其他外部组件，使项目轻量化

外部组件通常是不可控的，不利于系统结构的优化调整

世上只有两种程序员，一种是造过轮子的，另一种是没造过轮子的

class bitarray():
    def __init__(self, length, buffer: bytearray = None):
        self.length = length
        self.buffer = buffer or bytearray((length + 7) // 8)

    def __getitem__(self, item):
        if item < 0 or item >= self.length:
            raise IndexError("index out of range")

        index = item // 8
        offset = item % 8

        byte = self.buffer[index]
        mask = 128 >> offset
        return byte & mask == mask

    def __setitem__(self, item, value):
        if item < 0 or item >= self.length:
            raise IndexError("index out of range")

        index = item // 8
        offset = item % 8

        byte = self.buffer[index]
        if value:
            byte = byte | (128 >> offset)
        else:
            byte = byte & ~(128 >> offset)
        self.buffer[index] = byte


if __name__ == '__main__':
    bs = bitarray(1024 * 10)
    print(bs[0])

    bs[0] = True
    print(bs[0])

    bs[0] = False
    print(bs[0])

本文作者： killf
本文链接： http://www.killf.info/编程语言/Python/Python轻量级的bit数组/
版权声明： 本博客所有文章除特别声明外，均采用 Apache License 2.0 许可协议。转载请注明出处！