從numpy的3d數組中刪除2d子數組

Question

在numpy中，我有一個3d數組，我會ike刪除一些2d子數組。想想看這樣的：

r = range(27) 
arr = np.reshape(r, (3,3,3)) 

del = [[0,1,2],[0,0,2]] 
flatSeam = np.ravel_multi_index(del, arr.shape) 
arr = np.delete(arr, flatSeam) 

所以在最後我想有形狀(3,2,3)沒有從原來的數組中的元素00, 10, 22的數組。我的問題是我沒有使用ravel_multi_index，因爲我的索引是2d，數組形狀是3d，所以計算了錯誤的索引（上面的代碼也不會執行，因爲索引數組和形狀必須相同尺寸）。

你有什麼想法，我可以做到這一點？

Answer 1

下面是一個使用advanced-indexing的方法 -

# arr: Input array, rm_idx : 2-row list/array of indices to be removed 
m,n,p = arr.shape 
mask = np.asarray(rm_idx[1])[:,None] != np.arange(n) 
out = arr[np.arange(m)[:,None],np.where(mask)[1].reshape(m,-1)] 

或者，boolean-indexing -

out = arr.reshape(-1,p)[mask.ravel()].reshape(m,-1,p) 

少一點內存密集型的做法，我們儘量避免創建2Dmask -

vmask = ~np.in1d(np.arange(m*n),rm_idx[1] + n*np.arange(m)) 
out = arr.reshape(-1,p)[vmask].reshape(m,-1,p)