python - 两个相同形状的复杂多维数组中的条件检查

Question

我有两个 10x2x2 complex64 数组。我想找出哪些 2x2 数组中的一个或两个不是全为零：

import numpy
a = numpy.zeros((10,2,2), "complex64")
b = numpy.ones((10,2,2), "complex64")

empty_one_or_both = (a.reshape(10,4) != 0).all(axis=1) * (b.reshape(10,4) != 0).all(axis=1) # EDIT

目标是仅对非空对执行其他操作，例如：

numpy.sqrt(a[empty_one_or_both])

有没有更好的办法？

Answer 1

从 numpy 1.7 开始，您可以执行以下操作：

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

有关更多信息，请参阅文档，但如果您使用的是旧版本，我认为您必须重塑或执行 all(1).all(1)。