如何刪除在陣列就地項

Question

我用下面的線陣列的列表：

list_split_test_array = np.split(test_array, np.where(test_array == 0.)[0]) 

在每個分割爲0的數組，即我任意先前添加作爲佔位符的值。
這給了我一個陣列的列表，其中每個數組的第0個索引（除了第一個數組）都是0. 如何刪除這些0的就地，以獲得沒有佔位符0的數組列表？我說就地，因爲我寧願不迭代，添加刪除0每個數組到一個新的數組列表。

Answer 1

你可以使用一些預處理去除零，這裏有一個例子：

import numpy as np 
a = np.random.randint(0, 5, 100) 
idx = np.where(a == 0)[0] 

np.split(a, idx) 

輸出：

[array([4, 3]), 
array([0, 3, 1, 1]), 
array([0, 4]), 
array([0]), 
array([0, 1, 2, 4]), 
array([0, 4, 2, 2, 3, 1]), 
array([0, 2, 4, 3]), 
array([0, 2, 2, 4]), 
array([0, 3, 2, 1, 2, 4, 1, 2]), 
array([0]), 
array([0]), 
array([0, 1, 3]), 
array([0]), 
array([0, 1, 3, 2, 4]), 
array([0, 3, 3]), 
array([0, 3]), 
array([0, 3, 4, 1, 2, 3, 3, 4, 3, 4, 2, 1]), 
array([0]), 
array([0, 3, 2]), 
array([0]), 
array([0, 4, 1, 4, 2, 3, 1]), 
array([0, 4]), 
array([0]), 
array([0]), 
array([0, 3, 4, 2, 3]), 
array([0, 4, 4, 3, 3, 3, 1]), 
array([0, 2, 4, 1]), 
array([0, 4, 2, 2]), 
array([0])] 

撤除a全部爲零，並計算新的索引：

idx2 = idx - np.arange(len(idx)) 
b = np.delete(a, idx) 
np.split(b, np.unique(idx2)) 

這是輸出：

[array([4, 3]), 
array([3, 1, 1]), 
array([4]), 
array([1, 2, 4]), 
array([4, 2, 2, 3, 1]), 
array([2, 4, 3]), 
array([2, 2, 4]), 
array([3, 2, 1, 2, 4, 1, 2]), 
array([1, 3]), 
array([1, 3, 2, 4]), 
array([3, 3]), 
array([3]), 
array([3, 4, 1, 2, 3, 3, 4, 3, 4, 2, 1]), 
array([3, 2]), 
array([4, 1, 4, 2, 3, 1]), 
array([4]), 
array([3, 4, 2, 3]), 
array([4, 4, 3, 3, 3, 1]), 
array([2, 4, 1]), 
array([4, 2, 2]), 
array([], dtype=float64)]