Python循环遍历列表并根据条件追加

Question

我有 3 个非常长的列表(长度为 15,000)。比方说这三个列表是:

A    B    C
0    2    3
0    4    5
0    3    3
1    2    6
1    3    5
0    2    7
1    8    8

我想获取 B 和 C 的所有那些值，这些值是 A 的相应索引为 0 的地方。例如，如果 A[i] == 0然后我想添加 B[i]至 listB_0和 C[i]至 listC_0 .

我试过了

listB_0 = []
listC_0 = []

for a,b,c in zip(A,B,C):
    if a == 0:
        listB_0.append(B)
        listC_0.append(C)

但这似乎让 Python 进入了一个永无止境的循环，即使在 5 分钟后，我看到程序仍在运行。

我最终想要的是，例如 listA = 0 的 listB 和 listC

listB_0 = [2,4,3,2]
listC_0 = [3,5,3,7] 

实现这一目标的正确方法是什么？

Answer 1

Brobin已经在他的评论中指出了这一点:不是 b 或 c，而是 append 了整个列表 B 或 C .

这应该有效:

A = [0, 0, 0, 1, 1, 0, 1]
B = [2, 4, 3, 2, 3, 2, 8]
C = [3, 5, 3, 6, 5, 7, 8]

listB_0 = []
listC_0 = []

for a, b, c in zip(A,B,C):
    if a == 0:
        listB_0.append(b)
        listC_0.append(c)

print listB_0
print listC_0

>>> 
[2, 4, 3, 2]
[3, 5, 3, 7]