迭代numpy.ma数组，忽略掩码值

Question

我想在np.ma.ndarray。

中仅迭代未屏蔽的值

以下内容：

import numpy as np
a = np.ma.array([1, 2, 3], mask = [0, 1, 0])
for i in a:
    print i

我明白了：

1
--
3

我想得到以下内容：

1
3

好像np.nditer()可能是要走的路，但我找不到任何可能指定此内容的flags。我怎么能这样做？谢谢！

Answer 1

您想使用a.compressed()

import numpy as np
a = np.ma.array([1, 2, 3], mask = [0, 1, 0])
for i in a.compressed():
    print i

给出：

1
3