python - 删除numpy数组中的屏蔽元素

Question

我有一些包含屏蔽元素的数组(来自Numpy.MaskedArray)，例如

data = [0,1,masked,3,masked,5,...]

面具不遵循规则模式的地方。

我想遍历数组并简单地删除所有被屏蔽的元素以结束:

data = [0,1,3,5,...]

我试过这样的循环:

for i in xrange(len(data)):
    if np.ma.is_masked(data[i]):
        data.pop(i)

但我收到错误:赋值前引用的局部变量“数据”

我必须创建一个新数组并添加未屏蔽的元素吗？或者是否有可以自动执行此操作的 MaskedArray 函数？我看过文档，但对我来说并不明显。

谢谢!

Answer 1

data.compressed()是你要找的功能