Python itertools.batched 怎样处理末尾不足一组的数据

处理列表、日志或消息流时，经常需要把输入按固定数量拆成一组。itertools.batched() 的关键点是：默认不会因为最后一组不足而丢弃数据，也不会自动补空值，而是返回一个更短的元组。

官方文档：https://docs.python.org/3/library/itertools.html#itertools.batched

例如 7 个元素按 3 个一组，默认结果是两个长度为 3 的元组和一个长度为 1 的元组。如果业务必须每组都完整，再使用 strict=True，让不完整尾组变成明确的 ValueError。

默认模式会保留末尾不足一组的数据

batched(iterable, n) 返回的是迭代器，每次最多取 n 个元素组成元组。输入长度不能整除 n 时，最后剩下的元素仍然会被包装进一组，只是这组长度小于 n。

from itertools import batched

records = ["A", "B", "C", "D", "E", "F", "G"]

# 每组三条；默认模式保留最后不足三条的数据
groups = list(batched(records, 3))
print(groups)
# [('A', 'B', 'C'), ('D', 'E', 'F'), ('G',)]

这里的 list() 只是为了把迭代器一次性展开，便于观察结果。实际处理大输入时，可以直接用 for group in batched(records, 3)，这样每次只消费当前批次附近的数据，不必先复制出完整的二维列表。

为什么最后一组不会自动补齐

这个接口的职责是“按最多 n 个元素分组”，不是填充器。官方文档给出的等价逻辑会从输入中切出最多 n 个元素；只要这一批非空，就把它作为元组产出。因此最后只剩一个元素时，得到 ("G",) 是正常语义。

这也意味着不要用“最后一组长度不等于 n”来判断接口失败。需要逐条处理剩余数据时，直接接受短元组即可：

from itertools import batched

records = range(7)
for index, group in enumerate(batched(records, 3), start=1):
    # 尾组可以是短元组，业务代码应按真实长度处理
    print(f"第{index}组：{group}，数量={len(group)}")

如果下游协议明确要求固定三列，则那是业务层的补齐规则，需要先决定补充值是否安全，再单独实现；batched() 不会替你把缺失元素伪造成真实数据。

必须整组时用 strict=True

当批次代表一次完整提交、固定大小的向量或不可拆分的协议帧时，短尾可能应该被拒绝。Python 3.13 起可以传入 strict=True：完整批次照常返回，最后一批不足 n 个元素时抛出 ValueError。

from itertools import batched

records = ["A", "B", "C", "D"]
try:
    # strict=True 表示每一组都必须达到 n 个元素
    for group in batched(records, 3, strict=True):
        print(group)
except ValueError as exc:
    # 把不完整尾组转换成业务可识别的失败信息
    print(f"批次不完整：{exc}")

这个例子会先得到 ("A", "B", "C")，随后在发现剩余的 ("D",) 时抛出异常。不要把 strict=True 理解成“自动丢弃尾部”：它的作用是暴露不完整输入，让调用方决定重试、补齐或拒绝。

空输入、n 参数和业务选择

可以把常见边界记成下面这张表：

因此，分页展示、逐批写入且允许最后少几条时，用默认模式最自然；固定长度计算或必须整包提交时，用 strict=True 更安全。若运行环境低于 Python 3.12，标准库中没有这个接口；若需要 strict，还要确认环境至少是 Python 3.13。

常见疑问

末尾短元组会被补成 None 吗？

不会。默认模式只返回实际存在的元素；需要填充时，应由业务代码明确指定填充值和补齐条件。

strict=True 能忽略不完整尾组吗？

不能。它的语义是发现短尾就抛出 ValueError。如果要忽略短尾，应在默认模式下检查 len(group) == n 后再决定是否处理。

batched 会一次性读取整个列表吗？

不会。它按批次惰性消费输入；只有外层的 list() 或其他收集操作才会把所有结果保存到内存中。

记住一句话：允许尾批不足，就使用默认的 batched(iterable, n)；要求每组严格完整，就使用 strict=True 并捕获 ValueError。这样“保留数据”和“拒绝不完整输入”两种需求都能表达清楚。