如何快速简单打包4位有符号值？Python中[-8,7]整数高效字节打包方案

核心逻辑 #

取值范围在[-8,7]的整数可直接用4位补码表示，对应无符号4位值的范围是0-15（-8对应8，-1对应15，7对应7）。我们可以将每两个数打包进一个字节：

• 先把每个有符号数转成等效的4位无符号值（用x & 0xF快速完成补码转换）
• 每两个值组合为一个字节：第一个值左移4位作为高4位，第二个值作为低4位，通过按位或合并

高效实现代码 #

def pack_nibbles(nibbles):
    # 将所有有符号值转换为4位无符号等效值
    unsigned = [x & 0xF for x in nibbles]
    # 处理奇数长度的情况，自动补0（可根据需求调整补位规则）
    if len(unsigned) % 2 != 0:
        unsigned.append(0)
    # 按每两个值打包为一个字节，生成器表达式配合bytes()实现高效转换
    packed = bytes( (unsigned[i] << 4) | unsigned[i+1] for i in range(0, len(unsigned), 2) )
    return packed

关键细节说明 #

• 补码转换：x & 0xF是Python中快速获取4位补码无符号值的最优方式，无论输入正负，都会直接提取低4位二进制值，完美匹配4位补码的要求（例如-8 & 0xF结果为8，-1 & 0xF结果为15）。
• 效率优化：列表推导和生成器表达式避免了低效的显式循环累加，bytes()构造函数基于底层C实现，处理速度远快于手动字节拼接。
• 奇数长度兼容：若输入列表长度为奇数，代码会自动补0填充最后一个字节的低4位，你可根据实际场景调整补位逻辑（如补特定值或抛出异常）。

测试示例 #

# 测试包含正负值和奇数长度的输入
test_list = [-8, 7, -1, 0, 3]
packed_bytes = pack_nibbles(test_list)
print(packed_bytes)  # 输出: b'\x87\xf0\x30'

# 可选解包验证
def unpack_nibbles(packed):
    nibbles = []
    for b in packed:
        # 高4位转回有符号值
        high = (b >> 4)
        if high >= 8:
            high -= 16
        nibbles.append(high)
        # 低4位转回有符号值
        low = b & 0xF
        if low >= 8:
            low -= 16
        nibbles.append(low)
    # 移除补位的0（若原列表为奇数长度）
    if len(test_list) % 2 != 0:
        nibbles.pop()
    return nibbles

print(unpack_nibbles(packed_bytes))  # 输出: [-8, 7, -1, 0, 3]

内容的提问来源于stack exchange，提问作者astrophobia