有效地从nparray中移除零

Question

在我使用这个方法的时候：

data = np.array([[0, 0, 0, 0, 1, 2, 3, 4, 5, 0, 6, 0, 0], [0, 0, 0, 0, 1, 2,3, 4, 5, 0, 6, 0, 0]])
index = 0
idx = []
for img in range(len(data)):
    img_raw = np.any(data[img])
    if img_raw == 0.0:
        idx.append(index)
    index+=1
data = np.delete(data, idx, axis=0)

有人知道更好的方法吗？在

Answer 1

无论data是什么，Daniel都会回答1d数组，这在您的例子中似乎已经足够了。如果您的data数组是2d的，那么情况会变得稍微复杂一些，因为在不改变数组的维数的情况下无法删除0。在这种情况下，您可以使用mask-arrays 从你的考虑中去掉不想要的值，例如

import numpy as np
ma_data = np.ma.masked_equal(data,0)
print(ma_data)

任何计算，比如mean、std等等，都不要考虑屏蔽值。在

Answer 2

使用逻辑索引：

data = np.zeros(500)
data = data[data!=0]