我有一个很大的ndarray X (大概是(1e3,1e3,1e3)),我想做X的操作,包括和不包括第0轴的特定元素(对于第一和第二轴的每个元素)。也就是说,有(1e3,1e3)个元素我想(有时)屏蔽进去或屏蔽出去。
最简单的方法是构造一个掩码数组,
Z = np.zeros_like(X, dtype=bool)
# assume `inds` is some indexing array which will target
# the particular (1e3 x 1e3) elements I'm interested in
Z[inds] = True
Y = np.ma.masked_array(X, mask=Z)但这仅用于掩蔽数组就使用了额外的of内存。有没有办法在不构造掩码的第二个10^9元素数组的情况下做到这一点?例如,是否可以为掩码构造一个稀疏矩阵?
https://stackoverflow.com/questions/44416217
复制相似问题