脾气暴躁的外部平等比较有多个要素

时间:2019-04-24 19:36:31

标签: python numpy numpy-broadcasting

我有两个numpy数组,正在与它们进行外部相等比较

m = np.random.randint(4,size=(4,4,4))
ar = np.array([1,2,3])
x = np.equal.outer(ar,m)

如果我有兴趣测试arm的每个元素,此方法很好用。但是,如果我想一次性进行多个比较,例如元素1 or 2,该怎么办?一种解决方案可能是

x = np.equal.outer(1,m)|np.equal.outer(2,m)

这对我的示例来说很好,但是对于我的真实数据,我需要进行数十次相等性比较,这也取决于用户的输入,因此我正在寻找一种更好的方法来实现此目的。

0 个答案:

没有答案