如何從多維數組中刪除總和值等於0的元素？

Question

我有32×32個灰度級圖像的堆疊在多維數組masks如下：

(2560, 1, 32, 32) 

一些這些圖像中的包含非零值，如果我總結它們的值。

我想從多維數組中刪除其值總和等於零的元素。

這是我如何接近：

for idx in range(masks.shape[0]): 
    if np.sum(masks[idx]) == 0: 
     masks = np.delete(masks, idx) 

第一次迭代之後，其中一個零和元素被發現，從形狀masks變化

(2560, 1, 32, 32)

到

(2621439,)

爲什麼會發生這種情況，我該如何解決？我想要的只是減少第一維度，例如

(2120, 1, 32, 32)

Answer 1

你得到一個一個維數組，因爲你沒有爲np.delete提供軸參數，根據文檔：

如果axis是無，obj應用到平面數組。

而且它不是通過一個數組一個好主意，循環，同時刪除元素，這使得它很難保持，因爲對象的大小指數的軌跡正在發生變化：

對於您的情況：

masks = masks[masks.sum(axis=(1,2,3)) != 0] 

應該刪除軸的所有值之和爲1,2,3爲零的數組。