エレガントインデックス3次元配列への道とnumpyの

Question

使用して、非ゼロの軸を保つためには、私は、次の3次元配列を持っていると言う：L[0,:]ではゼロ列は常にL[1,:]にゼロの列が対応することで一致している

L=np.arange(18).reshape((2,3,3)) 
L[:,:,1] = 0; L[:,[0,1],:] = 0 

In []: L 
Out[]: 
array([[[ 0, 0, 0], 
     [ 0, 0, 0], 
     [ 6, 0, 8]], 

     [[ 0, 0, 0], 
     [ 0, 0, 0], 
     [15, 0, 17]]]) 

。

私は今、軸に沿って合計が0（行ゼロのを無視して等しい途中列を削除したい。私の現在の不器用なアプローチが

ことの少ない回り道で何

l=np.nonzero(L.sum(axis=1))[1] 

In []: L[:,:,l[:len(l)/2]] 
Out[]: 
array([[[ 0, 0], 
     [ 0, 0], 
     [ 6, 8]], 

     [[ 0, 0], 
     [ 0, 0], 
     [15, 17]]]) 

です？この

Answer 1

我々は最初の2つの軸に沿ってすべてゼロを探し、第3の軸からのものをマスクするためのものを使用することができます -

L[:,:,~(L==0).all(axis=(0,1))] 

また、~all()交換するany()を使用して - もっと上

L[...,~(L==0).all((0,1))] 
L[...,(L!=0).any((0,1))] 

- 私たちは:,:を交換しても、私たちのコンパクト版を与えるために引数axisをスキップするように省略記号表記...を使用することができます

L[:,:,(L!=0).any(axis=(0,1))] 

をNumPy配列の省略記号の意味はhereです。

---

質問の和の部分については、それは同様であろう -

L[...,L.sum((0,1))!=0]