Python 字节数组切片上的位掩码操作

Question

假设我有一个如下所示的字节数组：
mask = bytearray([0b0001, 0b0100, 0b0111]),
其中每个位代表一个特定的标志。我想像这样在掩码数组的切片中添加一个标志：
mask[0:2] = mask[0:2] | 0b1000
但我得到一个 TypeError :
TypeError: unsupported operand type(s) for |: 'bytearray' and 'int'
这样做最优雅的方式是什么？

也试过这个：
masks[0:2] = bytearray([bin(m | 0b0001) for m in masks[0:2]]) 有错误：
string must be of size 1

谢谢！

Answer 1

如果不求助于numpy，则无法直接完成，但您可以解决它：

mask[:2] = (b | 0b1000 for b in mask[:2])

print(list(map(bin, mask)))

得到你：

['0b1001', '0b1100', '0b111']

您的最后一次尝试已经接近尾声，但是当您真正想要原始整数值时，包装 bin 正在转换为字符串表示形式。