在两个具有相同形状的列表中查找实例相似

Question

我正在使用时间表数据。假设我有两个相等形状的列表，我需要找到两个列表在同一位置的数字大于零的实例。

为了分解它，

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

我们可以看到在四个位置中，两个列表的数字都大于0。还有其他实例，但是我确定如果我可以获得一个简单的代码，它所有的需求就是调整符号，我也可以在更大的规模。

我已经尝试过

len([1 for i in A if i > 0 and 1 for i in B if i > 0 ])

，但我认为它给我的答案是两种情况的产物。

Answer 1

既然您有一个 numpy tag：

A = np.array([1,0,2,0,4,6,0,5])
B = np.array([0,0,5,6,7,5,0,2])

mask = ((A>0)&(B>0))
# array([False, False,  True, False,  True,  True, False,  True])

mask.sum()
# 4

A[mask]
# array([2, 4, 6, 5])

B[mask]
# array([5, 7, 5, 2])

in Pure Python（可以推广到任何数量的列表）：

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

mask = [all(e>0 for e in x) for x in zip(A, B)]

# [False, False, True, False, True, True, False, True]

Answer 2

如果您想使用香草python，这应该是在做您想要的

l = 0
for i in range(len(A)):
    if A[i] > 0 and B[i] > 0:
        l = l + 1