python - 检查列表并相应地编辑第二个列表

Question

我有 3 个列表(相同的形状 - 大约 800 个元素)，值为 1 或 0。我想要实现的是一个新列表，其中包含该索引的其他 3 个列表中 1 的数量。

进一步解释...

# I have these
list1 = [1, 0, 1]
list2 = [1, 1, 1]
list3 = [0, 0, 1]

# I want this

new_list = [2, 1, 3]

我通过嵌套的 for 循环(即循环遍历每个列表并相应地编辑 new_list)来实现它。但是，我想知道是否存在更有效(尤其是在内存和资源方面)的方法。

Answer 1

这应该适合你

# I have these
list1 = [1, 0, 1]
list2 = [1, 1, 1]
list3 = [0, 0, 1]

[sum(i) for i in zip(list1, list2, list3)]

[2, 1, 3]