寻找numpy图像的重叠部分

Question

如果我有两个numpy数组：

a = np.array[[0,1,1],
             [0,1,0],
             [0,0,0]] 
b = np.array[[0,1,1],
             [0,0,1],
             [0,0,0]]

我该如何找到它们之间的“重叠”，以便输出为：

c =         [[0,1,1],
             [0,0,0],
             [0,0,0]]  

我有这个，但是否有更快的方法，因为我的数组很大？

c = a + b - 1
c[c<0] = 0

- kezzos

---

希望那里的答案有所帮助。 - Mathias711

Answer 1

如果像您的例子一样，数组中的值只有0或1，您可以使用按位“与”运算符&：

In [3]: a
Out[3]: 
array([[0, 1, 1],
       [0, 1, 0],
       [0, 0, 0]])

In [4]: b
Out[4]: 
array([[0, 1, 1],
       [0, 0, 1],
       [0, 0, 0]])

In [5]: c = a & b

In [6]: c
Out[6]: 
array([[0, 1, 1],
       [0, 0, 0],
       [0, 0, 0]])

- Warren Weckesser

---

谢谢。事实证明，c = a == b 实际上要快得多！ - kezzos

@kezzos，这是相同的结果吗？如果a和b的值匹配，c = a == b将返回true，而c = a＆b将在两者都为1时返回1。 - Chris Farr