首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >使用numpy掩码数组有效地使用内存

使用numpy掩码数组有效地使用内存
EN

Stack Overflow用户
提问于 2017-06-07 22:58:31
回答 0查看 906关注 0票数 3

我有一个很大的ndarray X (大概是(1e3,1e3,1e3)),我想做X的操作,包括和不包括第0轴的特定元素(对于第一和第二轴的每个元素)。也就是说,有(1e3,1e3)个元素我想(有时)屏蔽进去或屏蔽出去。

最简单的方法是构造一个掩码数组,

代码语言:javascript
复制
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元素数组的情况下做到这一点?例如,是否可以为掩码构造一个稀疏矩阵?

EN

回答

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/44416217

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档