python - 如何从 3d numpy 数组（28X28 行）中删除其他 3d numpy 数组（ Python ）中存在的所有行？

Question

我有两个包含 28x28 图像行的 3d numpy 数组。一个（“A”）具有（1000、28、28）的形状，另一个（“B”）具有（100、28、28）的形状，其中一些包含在后者中。如何过滤“A”以删除 B 中也包含的所有元素？

Answer 1

用理解索引第一个数组：

filtered = A[[0 if i in B else 1 for i in A]]

Answer 2

列表生成器：

new_A = [line for line in A if line not in B]

emm，有用吗？它应该比迭代更快。