我有两个numpy数组,正在与它们进行外部相等比较
m = np.random.randint(4,size=(4,4,4))
ar = np.array([1,2,3])
x = np.equal.outer(ar,m)
如果我有兴趣测试ar
中m
的每个元素,此方法很好用。但是,如果我想一次性进行多个比较,例如元素1 or 2
,该怎么办?一种解决方案可能是
x = np.equal.outer(1,m)|np.equal.outer(2,m)
这对我的示例来说很好,但是对于我的真实数据,我需要进行数十次相等性比较,这也取决于用户的输入,因此我正在寻找一种更好的方法来实现此目的。