2017-03-16 149 views
1

有沒有一種快速的方法(因此無循環)在numpy數組中找到每行的多數元素並從中創建一個新的數組?在numpy矩陣中獲取每行的大多數元素

例如,如果您有以下numpy的數組:

X = 
[[ 1. 1. 1.] 
[ 1. 0. 1.] 
[ 1. 0. 1.] 
[ 1. 1. 1.] 
[ 1. 0. 1.] 
[ 1. 0. 1.] 
[ 0. 0. 0.] 
[ 1. 1. 1.] 
[ 0. 0. 0.] 
[ 0. 0. 0.] 
[ 0. 0. 0.]] 

你可以做get_majority(X),這將輸出

[ 1. 1. 1. 1. 1. 1. 0. 1. 0. 0. 0.] 

我試着通過遍歷矩陣,並使用這樣一個Countercollections,但這對大矩陣非常緩慢,所以我想找到一個矢量化的方式來做到這一點。

回答

相關問題