遍历 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