python - 过滤掉两个/一个/两个都不为非零的矩阵元素？

Question

我有两个 numpy 数组 A 和 B。我想创建数组 Ap 和 Bp 这样 Ap 和 Bp 都是 A< 的元素 和 B，其中 A 或 B 中至少有一个为非零，或者两者都不为零。一个例子可能是:

A = [ 1.1, 0.0, 3.1, 4.1, 5.1, 0.0, 0.0, 8.1 ]
B = [ 0.0, 2.2, 3.2, 4.2, 0.0, 0.0, 7.2, 8.2 ]

然后在我要生成的一个函数中:

Ap = [ 1.1 0.0 3.1 4.1 5.1 0.0 8.1 ]
Bp = [ 0.0 2.2 3.2 4.2 0.0 7.2 8.1 ]

在另一个我想生成:

Ap = [ 3.1 4.1 8.1 ]
Bp = [ 3.2 4.2 8.1 ]

现在我只是浏览并查看每个元素，但我觉得应该有更好/更快的方法。

更新:

我有一个调用 (A,B) 的函数，它们是 [nx,ny,3] 大小的 numpy 数组。然后我调用以下内容:

tA = np.copy(A)
tB = np.copy(B)

tA = tA.flatten()
tB = tB.flatten()

Aeq0 = tA==0
Beq0 = tB==0

Ano0 = A!=0
Bno0 = B!=0

As = tA[ Ano0 | Bno0 ]
Bs = tB[ Ano0 | Bno0 ]

它死于 As = tA[ Ano0 | Bno0]

Answer 1

您可以使用屏蔽索引来实现这一点。

A = np.asarray(A)
B = np.asarray(B)

ind1 = A!=0
ind2 = B!=0

然后，实现第一种情况(|是一个“或”运算符):

case1 = ind1 | ind2 
Ap = A[case1]
Bp = A[case1]

而对于第二种情况(& 是一个“与”运算符):

case2 = ind1 & ind2 
Ap = A[case2]
Bp = A[case2]