python – 有效地从nparray中删除零

2023年6月3日 669次阅读

在我使用这种方法的那一刻：

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)

有人知道更好的方法吗？

最佳答案 无论数据是什么,Daniel都会回答1d阵列,这在你的情况下似乎已经足够了.如果你的数据数组是2d,那么事情变得有点复杂,因为你不能在不改变数组尺寸的情况下删除0.在这种情况下,您可以使用
mask-arrays

 从您的考虑中删除不需要的值,例如

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

任何计算,比如mean,std等,都不会考虑屏蔽值.

点赞