python二維數組，切片/從布爾掩模刪除元素，並保持結果爲一個二維數組

Question

我有一個關於使用布爾掩碼掩蓋/切片2D numpy數組的問題。在這裏，我有一個np二維數組x，它是

x=np.tile(np.arange(10).reshape(10,1),(1,10)) 
[[0 0 0 0 0 0 0 0 0 0] 
[1 1 1 1 1 1 1 1 1 1] 
[2 2 2 2 2 2 2 2 2 2] 
[3 3 3 3 3 3 3 3 3 3] 
[4 4 4 4 4 4 4 4 4 4] 
[5 5 5 5 5 5 5 5 5 5] 
[6 6 6 6 6 6 6 6 6 6] 
[7 7 7 7 7 7 7 7 7 7] 
[8 8 8 8 8 8 8 8 8 8] 
[9 9 9 9 9 9 9 9 9 9]] 

x是一個10乘10陣列，其每列都是相同的。如果有更好的方法來產生這個，你能指出一下嗎？謝謝。再有就是一個布爾數組，

y=(x!=np.arange(10)) 
[[False True True True True True True True True True] 
[ True False True True True True True True True True] 
[ True True False True True True True True True True] 
[ True True True False True True True True True True] 
[ True True True True False True True True True True] 
[ True True True True True False True True True True] 
[ True True True True True True False True True True] 
[ True True True True True True True False True True] 
[ True True True True True True True True False True] 
[ True True True True True True True True True False]] 

現在，我想要得到的結果數組一樣，

[[1 0 0 0 0 0 0 0 0 0] 
[2 2 1 1 1 1 1 1 1 1] 
[3 3 3 2 2 2 2 2 2 2] 
[4 4 4 4 3 3 3 3 3 3] 
[5 5 5 5 5 4 4 4 4 4] 
[6 6 6 6 6 6 5 5 5 5] 
[7 7 7 7 7 7 7 6 6 6] 
[8 8 8 8 8 8 8 8 7 7] 
[9 9 9 9 9 9 9 9 9 8]] 

在X，其相應的位置Y中的元素已經False值被刪除，而新的2D數組被返回。那麼我怎麼能得到這種情況呢？

非常感謝您的幫助。

Answer 1

x[y]從x中選擇值，其中y爲真。然而，數組x[y]是一維的。您可以使用reshape(9,-1)將其重新塑造成9行（但無論多少列有意義）。爲了獲得理想的結果，您必須從左向右翻轉陣列：

In [81]: np.fliplr(x[y].reshape(9,-1)) 
Out[84]: 
array([[1, 0, 0, 0, 0, 0, 0, 0, 0, 0], 
     [2, 2, 1, 1, 1, 1, 1, 1, 1, 1], 
     [3, 3, 3, 2, 2, 2, 2, 2, 2, 2], 
     [4, 4, 4, 4, 3, 3, 3, 3, 3, 3], 
     [5, 5, 5, 5, 5, 4, 4, 4, 4, 4], 
     [6, 6, 6, 6, 6, 6, 5, 5, 5, 5], 
     [7, 7, 7, 7, 7, 7, 7, 6, 6, 6], 
     [8, 8, 8, 8, 8, 8, 8, 8, 7, 7], 
     [9, 9, 9, 9, 9, 9, 9, 9, 9, 8]])