python – 有效地从nparray中删除零

发布时间：2020-12-20 12:03:36 所属栏目：Python 来源：网络整理

导读：在我使用这种方法的那一刻： data = np.array([[0,1,2,3,4,5,6,0],[0,0]])index = 0idx = []for img in range(len(data)): img_raw = np.any(data[img]) if img_raw == 0.0: idx.append(index) index+=1data = np.delete(data,idx,axis=0) 有人知道更好的方

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

data = np.array([[0,1,2,3,4,5,6,0],[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等,都不会考虑屏蔽值.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!