如何实现适配两种参数的for循环解包统一逻辑？

你当前的写法存在两个问题：一是用if-else拆分循环导致代码冗余，二是当t3为None时，zip会因为参数不是可迭代对象抛出TypeError。下面提供两种统一循环的实现思路，解决这两个问题：

方案1：动态构建可迭代参数列表 #

通过过滤掉None值，动态生成zip的参数列表，然后在循环中统一处理解包后的元素：

def func(task, t1, t2, t3):
    # 收集需要参与zip的可迭代对象，排除None
    iterables = [t1, t2]
    if t3 is not None:
        iterables.append(t3)
    
    for items in zip(*iterables):
        # 统一提取基础元素
        t1_item, t2_item = items[0], items[1]
        t3_item = items[2] if len(items) == 3 else None
        
        # ----------------------
        # 这里写你的通用"do something"逻辑
        # 比如根据task或t3_item是否为None处理差异
        print(f"t1: {t1_item}, t2: {t2_item}", end="")
        if t3_item is not None:
            print(f", t3: {t3_item}")
        else:
            print()
        # ----------------------

方案2：用itertools.repeat填充空值 #

当t3为None时，用repeat(None)生成一个无限迭代器（zip会自动以最短的t1/t2长度为准截断），这样可以直接统一用三元组解包：

from itertools import repeat

def func(task, t1, t2, t3):
    # 处理t3的迭代器：非None则用原对象，否则生成重复的None
    t3_iter = t3 if t3 is not None else repeat(None)
    
    # 统一循环解包，无需拆分
    for t1_item, t2_item, t3_item in zip(t1, t2, t3_iter):
        # ----------------------
        # 这里写你的通用"do something"逻辑
        print(f"t1: {t1_item}, t2: {t2_item}", end="")
        if task != 'abc':
            print(f", t3: {t3_item}")
        else:
            print()
        # ----------------------

注意点 #

• 避免循环变量和函数参数同名：原代码中循环里的t1, t2会覆盖函数参数的t1, t2，这会导致后续逻辑出错，所以示例中改用t1_item这类命名。
• 两种方案都实现了循环逻辑的复用，无需拆分if-else，同时解决了t3为None时的报错问题。

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