计算每行中numpy数组中的trues数量?

时间:2017-08-05 07:13:44

标签: 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]]

并且返回值应该是:

[10
 15
 8
 ..., 
 11
 10
 12]

This question asks about how to do it for the whole array但我如何为每一行做到这一点?

1 个答案:

答案 0 :(得分:5)

我们可以通过向axis函数提供sum参数来执行此操作:

arr.sum(axis=1)