2016-12-15 46 views
1

我遇到了一个问题,有人想生成list3 = [1, 0, 1, 0, 0]list1 = [0, 1, 0, 0, 1]和list2 = [0, 1, 0, 1, 1]替代列表理解

如果有人会zip(list1, list2)因此只有zip(0, 0)将产生1list3

我想表达一个列表理解后续代码:

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

我可以通过这个list解析得到的结果相同:

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

我想知道是否有更直接地表达列表理解中的初始代码块的if/else条件的方式。我可以理解这个问题可能被认为是多余的,因为if/else条件自然是组成的,但发布的替代方案可能对我非常有用。

回答

5

你可以使用一个条件表达式:

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?

另一种方法是逻辑分解出成一个函数,并调用从列表解析的功能。这将允许逻辑使用任何流控制语句(条件语句,循环等)。

1
L1,L2 = [0, 1, 0, 0, 1], [0, 1, 0, 1, 1] 
# map and lambda 
L3 = map(lambda x: 1 if not sum(x) else 0, zip(L1, L2)) 
print L3 
# [1, 0, 1, 0, 0]