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

Question

我有一个形状为 (1000, 12, 30) 的 3d 数组，我有一个形状为 (12, 30) 的 2d 数组列表，我想要做的是检查这些 2d 数组是否存在于 3d 数组中。 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)))