減少numpy中的稀疏矩陣

Question

看來我在這裏錯過了一些非常基本的東西。 我有一個大型的方陣，其中大部分爲零。我想要的是將其縮減爲包含具有非零條目的所有行和列的矩陣。 例如：

1 1 0 1 
1 1 0 1 
0 0 0 0 
1 1 0 1 

應減少到：

1 1 1 
1 1 1 
1 1 1 

有一個快速的方法來做到這一點？

Answer 1

怎麼是這樣的：

>>> arr 
array([[ 1., 1., 0., 1.], 
     [ 1., 1., 0., 1.], 
     [ 0., 0., 0., 0.], 
     [ 1., 1., 0., 1.]]) 

>>> mask = (arr==0) 

arr = arr[~np.all(mask,axis=0)] 
arr = arr[:,~np.all(mask,axis=1)] 
>>> arr 
array([[ 1., 1., 1.], 
     [ 1., 1., 1.], 
     [ 1., 1., 1.]])