在每行的numpy数组中计算正确的数目?

aro*_*oma 3 python numpy

我有一个numpy数组,我想返回true每一行的值计数。

例如我有一个numpy数组:

[[False False False ..., False False False]
 [False False False ..., False False False]
 [False False False ..., False  True False]
 ..., 
 [False False False ..., False False False]
 [ True False  True ...,  True  True  True]
 [False False False ..., False False False]]
Run Code Online (Sandbox Code Playgroud)

返回值应类似于:

[10
 15
 8
 ..., 
 11
 10
 12]
Run Code Online (Sandbox Code Playgroud)

这个问题询问如何对整个数组执行操作,但是如何对每一行执行操作呢?

aro*_*oma 7

我们可以通过为函数提供axis参数来简单地做到这一点sum

arr.sum(axis=1)
Run Code Online (Sandbox Code Playgroud)