2017-08-05 61 views
2

我有一個numpy數組,我想返回每行的true值的計數。計算每行中numpy數組中的真實數量?

,比如我有一個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但我怎麼做的每一行?

+2

你檢查的文檔的任何解決方案在這個問題的答案建議? ['numpy.sum'](https://docs.scipy.org/doc/numpy/reference/generated/numpy.sum.html)和['numpy.count_nonzero'](https://docs.scipy。 org/doc/numpy/reference/generated/numpy.count_nonzero.html)帶一個可選的'axis'參數。 – user2357112

+0

我不知道用普通的python做的numpy,就像'r = [len(filter(None,a))for a source]'''source'是你的原始列表 – shevron

+0

@ user2357112謝謝我發現它出。 – ash

回答

4

我們可以簡單地做到這一點通過提供axis參數的sum功能:

arr.sum(axis=1)