python - 如何对一个二维numpy数组的所有列进行逻辑运算

Question

假设我有以下 2D NumPy 数组，由四行三列组成:

>>> a = numpy.array([[True, False],[False, False], [True, False]])
>>> array([[ True, False],
       [False, False],
       [ True, False]], dtype=bool)

生成包含逻辑或所有列(如 [True, False])的 1D 数组的有效方法是什么？

我在网上搜索了一下，发现有人引用sum(axis=)来计算sum。

不知道有没有类似的逻辑运算方式？

Answer 1

是的，有。使用任何:

>>> a = np.array([[True, False],[False, False], [True, False]])
>>> a
array([[ True, False],
       [False, False],
       [ True, False]], dtype=bool)
>>> a.any(axis=0)
array([ True, False], dtype=bool)

请注意当您将参数 axis 更改为 1 时会发生什么:

>>> a.any(axis=1)
array([ True, False,  True], dtype=bool)
>>> 

如果你想要逻辑和使用all:

>>> b.all(axis=0)
array([False, False], dtype=bool)
>>> b.all(axis=1)
array([ True, False, False], dtype=bool)
>>> 

另请注意，如果您省略 axis 关键字参数，它适用于每个元素:

>>> a.any()
True
>>> a.all()
False