检查Python中的3d数组中是否存在2d数组？

Question

我有一个带有shape的3d数组(1000, 12, 30)，并且有一个2d数组的shape列表(12, 30)，我想要做的是检查3d数组中是否存在这些2d数组。Python中有一种简单的方法可以做到这一点吗？我尝试了关键字，in但是没有用。

Answer 1

有一种方法numpy，你可以用np.all

a = np.random.rand(3, 1, 2)
b = a[1][0]
np.all(np.all(a == b, 1), 1)
Out[612]: array([False,  True, False])

来自 bnaecker 的解决方案

np.all(a == b, axis=(1, 2))

如果只想检查是否退出

np.any(np.all(a == b, axis=(1, 2)))