仅当单独的布尔列表中的元素为True时,才会对python列表中的元素求和

Question

我有两个python列表,

A = [ 1, 2, 3, 4, 5 ]
B = [ True, False, False, True, True ]

列表A和B的长度相同.

我想只总结A中与B中的True元素相对应的元素.我知道我可以用以下方法做到:

sum([A[x] for x in xrange(len(A)) if B[x]])

但我想知道是否有一个更优雅的解决方案,不涉及循环每个列表中的元素？

Answer 1

使用itertools.compress:

>>> from itertools import compress
>>> sum(compress(A, B))
10

itertools.compress链接页面上描述了实现.这是短期和简单,所以你不要有进口itertools ¹:

>>> sum(a for a, b in zip(A, B) if b)
10

_{¹ OTOH,itertools.compress用C实现,因此应该更快}