python - 按数组数组中的列计算 bool 值

Question

我有一个包含 bool 值的数组:

[[False False  True ..., False  True False]
 [False  True  True ...,  True False  True]
 [False False False ...,  True  True False]
 ..., 
 [False False False ..., False False False]
 [ True  True  True ...,  True  True  True]
 [ True  True  True ...,  True  True  True]]
<type 'numpy.ndarray'>

以下代码计算行

中的真值

results = []
for r in my_array:
   results.append(np.sum(r))

如何按列计算 bool 值的数量？

Answer 1

numpy.sum支持对跨多个轴的数组求和。对列使用第 0 轴，对行使用第 1 轴。

>>> arr = np.ndarray(shape=(3, 4), dtype=bool)
>>> arr
array([[False,  True, False,  True],
       [False, False, False,  True],
       [False, False, False, False]], dtype=bool)
>>> np.sum(arr, axis=0)
array([0, 1, 0, 2])
>>> np.sum(arr, axis=1)
array([2, 1, 0])