如何为生成笛卡尔积的Python递归生成器添加仅返回末尾两数为偶数结果的逻辑？

核心实现思路 #

你原有的递归生成器逻辑本身没有问题，不建议在递归的遍历分支中提前插入判断条件。很多人一开始改这类递归函数的时候，总想着在每一层遍历的时候就提前过滤，但递归过程里拼出来的都是半截的子元组——比如处理到输入列表倒数第二层的时候，拼出来的元组只有1个元素，根本看不到完整结果的最后两位，提前判断很容易出现逻辑错位。
正确的插入点应该放在完整元组组装完成的位置：也就是拼好(a,) + prod得到完整的out元组之后，先校验是否满足过滤规则，满足条件再向上抛出，不满足就直接跳过当前循环。

基础版修改示例 #

直接在原有代码的yield之前加校验逻辑，代码如下：

def product(ar_list):
    if not ar_list:
        yield ()
    else:
        for a in ar_list[0]:
            for prod in product(ar_list[1:]):
                out = (a,) + prod
                # 校验逻辑：最后两个元素均为偶数
                if len(out) >= 2:
                    second_last, last = out[-2], out[-1]
                    if second_last % 2 == 0 and last % 2 == 0:
                        yield out

ar_list=[[1,2],[3,4],[5,6,7]]              
print(list(product(ar_list)))
# 输出结果：[(2, 4, 6)]

通用可扩展的过滤写法 #

如果需要频繁调整过滤规则，可以把过滤逻辑抽成独立的函数，作为参数传入生成器，避免每次改动递归函数的内部代码：

def product(ar_list, filter_func=None):
    if not ar_list:
        yield ()
    else:
        for a in ar_list[0]:
            for prod in product(ar_list[1:], filter_func):
                out = (a,) + prod
                # 无自定义过滤则直接放行，否则执行外部传入的校验逻辑
                if filter_func is None or filter_func(out):
                    yield out

# 自定义过滤规则：仅保留最后两个元素为偶数的元组
def keep_even_last_two(tup):
    if len(tup) < 2:
        return False
    return tup[-1] % 2 == 0 and tup[-2] % 2 == 0

ar_list=[[1,2],[3,4],[5,6,7]]              
print(list(product(ar_list, keep_even_last_two)))

常见踩坑提示 #

• 别在递归调用下层函数前提前判断元素属性：比如遇到当前层的元素就判断是不是偶数，再决定要不要往下递归，这种逻辑和“取元组最后两个元素判断”的需求完全错位。一旦输入列表的长度变化，比如多嵌套一层子列表，原来判断的位置就不是最后两位了，会出现完全不符合预期的错误过滤。
• 校验前必须先判断元组长度：如果输入列表的元素个数不足2个，直接取倒数第二个元素会触发索引越界，提前做长度校验可以兼容不同长度的输入场景，不会因为边界情况报错。

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