Python字典中每第3个bytearray切片后批量比较的简洁实现

问题背景 #

你拥有一个值为bytearray类型的Python字典test_bytearray_dict，需要对字典中每第3个元素（即键为test_array0、test_array3、test_array6、test_array9、test_array12、test_array15的元素）的bytearray执行[3:]切片操作，并验证这些切片后的结果是否全部相等。此前的多等式判断写法繁琐，需要更简洁的实现方式。

简洁解决方案 #

方法1：使用all()函数批量验证 #

通过生成目标键的序列，提取对应切片后，用all()函数判断所有切片是否与第一个切片相等：

# 生成需要检查的目标键（步长为3，从0到15）
target_keys = [f"test_array{i}" for i in range(0, 16, 3)]
# 获取第一个切片作为基准
base_slice = test_bytearray_dict[target_keys[0]][3:]
# 验证所有切片是否与基准相等
if not all(test_bytearray_dict[key][3:] == base_slice for key in target_keys):
    print('error: 切片结果不完全相等')

方法2：利用可哈希类型去重验证 #

由于bytearray不可直接存入集合，可先转换为bytes类型，通过集合的唯一性来判断切片是否全部一致：

target_keys = [f"test_array{i}" for i in range(0, 16, 3)]
# 将切片转为bytes后存入集合，去重
unique_slices = set(bytes(test_bytearray_dict[key][3:]) for key in target_keys)
# 若集合长度不为1，说明存在不同的切片
if len(unique_slices) != 1:
    print('error: 切片结果不完全相等')

更紧凑的单行写法 #

如果不需要保留中间变量，可直接在条件判断中生成序列：

if not all(test_bytearray_dict[f"test_array{i}"][3:] == test_bytearray_dict["test_array0"][3:] 
           for i in range(0, 16, 3)):
    print('error:')

方案优势 #

• 扩展性强：后续需要调整目标键的范围或步长时，仅需修改range()的参数即可
• 代码简洁：避免了手动编写大量等式判断，可读性和维护性更高
• 效率高效：all()函数会在遇到第一个不相等的元素时立即停止检查，无需遍历全部元素

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