Pythonでブールインデックスを作成した後の入力の形状を維持する

Question

私は以下の問題に助けが必要です。

Suppose X = [1 3 0 8 
      1 4 6 0 
      2 0 7 8 ] 
mask = (X != 0) 

mask = [ T T F T 
     T T T F 
     T F T T] 
X1 = X[(mask,np.newaxis)] 

その出力X1形状（9,1）

であるが、私はX1は、（3,3）のようにしたい、すなわち、マスクされたエントリを除き、Xと同様の形状を維持します。

X1 = [1 3 8 
    1 4 6 
    2 7 8 ] 

誰かが私を助けることができますか？ありがとうございました。

Xのすべての行にはゼロが含まれ、reshape（）は使用しません。ここでの作業は

X= np.array([[1,3,0,8],[1,4,6,0],[2,0,7,8]]) 
mask = (X!=0) 
X1=X[(mask,np.newaxis)] 

出力Xは形状（9,1）です。 X1が（3,3）のようになる方法はありますか？

Answer 1

あなたの質問には正しい構文も含まれていないので、Pythonで簡単に何かを始めたいと思うかもしれません。私はこれがちょうど擬似コードの試みだったと思っています。しかし、ここであなたが望むマスクを行うためのコードがあります。

import numpy as np 
X = np.array([1, 3, 0, 8, 1, 4, 6, 0, 2, 0, 7, 8]) 
indicies_we_want = np.where(X > 0) # Results in an array containing the indicies of X we want to keep 
result = np.take(X, indicies_we_want) # Filter by these indicies 
result = result.reshape(3, 3) # Reshape to desired result 
print result 

このコードはかなり圧縮されている可能性がありますが、わかりやすくするために各ステップをご説明したいと思います。

コメントセクションで指摘されているように、通常、リシェープは良いアイデアではありません。なぜなら、0を除外して9個の要素が残っていることがわかっていない限りです。あなたが記述した場合、私たちは確かにこれを知っていますが、特定の配列ではそれほど多くはありません。変形と

Answer 2

In [173]: x=[[1,3,0,8],[1,4,6,0],[2,0,7,8]] 
In [174]: xa=np.array(x) 

溶液：

In [175]: xa[xa!=0].reshape(3,3) 
Out[175]: 
array([[1, 3, 8], 
     [1, 4, 6], 
     [2, 7, 8]]) 

溶液変形なし：明らか

In [176]: np.array([i[i!=0] for i in xa]) 
Out[176]: 
array([[1, 3, 8], 
     [1, 4, 6], 
     [2, 7, 8]]) 

両方が行ごとに1つだけ削除さに依存します。

共通の列は削除されません。コード内の何も、結果が変形可能であるということを基礎とするnumpyに伝えません。したがって、ブーリアンインデックスはフラット化された配列で動作します。

In [177]: xa[xa!=0] 
Out[177]: array([1, 3, 8, 1, 4, 6, 2, 7, 8]) 
In [178]: xa.flat[xa.flat!=0] 
Out[178]: array([1, 3, 8, 1, 4, 6, 2, 7, 8]) 

私は余分な0を投げてしまい、このインデックス付けは同じように動作します。それを3x3に改造する努力は失敗するでしょう。

---

基礎となるデータ・バッファは、1D平坦であり、それが唯一の理由shapeとstriding属性の2Dとして表示されることことに留意してください。要素を選択（またはスキップ）するとコピーが作成され、1次元コピーは2次元コピーと同じくらい簡単で高速です。 reshapeはデータバッファを変更せず、shape属性のみを変更します。