python中in的用法数组_numpy数组的“In”操作符？

您可以比较equality的输入数组，这将对a中最后两个轴上每个位置的所有元素与第二个数组中相应位置的元素进行{a1}比较。这将产生一个匹配的布尔数组，在这个数组中，我们检查最后两个轴上的^{}匹配，最后检查{a3}匹配，如下-((a==b).all(axis=(1,2))).any()

样本运行

1)输入：

^{pr2}$

2)广播元素比较：In [70]: a==b

Out[70]:

array([[[False, False, True],

[False, False, True]],

[[ True, True, True],

[ True, True, True]],

[[False, True, True],

[False, True, True]],

[[False, False, True],

[False, False, True]],

[[ True, False, True],

[ True, False, True]]], dtype=bool)

3)ALL匹配最后两个轴，最后ANY匹配：In [71]: (a==b).all(axis=(1,2))

Out[71]: array([False, True, False, False, False], dtype=bool)

In [72]: ((a==b).all(axis=(1,2))).any()

Out[72]: True

对a中的c执行类似的步骤-In [73]: c

Out[73]:

array([[300, 200, 0],

[ 0, 100, 100]])

In [74]: ((a==c).all(axis=(1,2))).any()

Out[74]: False