Skip to main content
 首页 » 编程设计

python之替代列表理解

2025年05月04日16daizhj

我遇到了一个问题,有人想生成 list3 = [1, 0, 1, 0, 0]对于 list1 = [0, 1, 0, 0, 1 ] 和 list2 = [0, 1, 0, 1, 1] .

如果有人会 zip(list1, list2)因此只有zip(0, 0)会生成 1对于 list3 .

我想用列表理解来表达后续代码:

L3 = [] 
for x in zip(L1, L2): 
    if sum(x) == 0: 
        L3.append(1) 
    else: 
        L3.append(0) 
print L3  

我可以通过使用这个列表理解得到相同的输出:

print [int(sum(x)==0) for x in zip(L1, L2)] 

我想知道是否有一种方法可以更字面地表达列表理解中初始代码块的 if/else 条件。我可以理解这个问题可能被认为是多余的,因为 if/else 条件自然包含在内,但发布的替代方案对我来说可能非常有用。

请您参考如下方法:

您可以使用条件表达式:

print [1 if sum(x) == 0 else 0 for x in zip(list1, list2)] 

这里,1 if sum(x) == 0 else 0直接表达了条件。

有关详细信息,请参阅 Does Python have a ternary conditional operator?

另一种方法是将逻辑分解成一个函数,然后从列表理解中调用该函数。这将允许逻辑使用任何流程控制语句(条件语句、循环等)。