使用numpy数组掩码时，只有一列

Question

我只想屏蔽2D数组strut[i].rpsi中的col 0。下面的作品可以正确屏蔽第0列，但也可以屏蔽第1列。

mask = ~(strut[i].rpsi <= time.time() - apsiTimeWindow)

上面的输出是：

[[False False]
 [True False]
 [True False]]

我需要这样的输出：

[[False True]
 [True True]
 [True True]]

甚至更好：

[[False]
 [True]
 [True]]

Answer 1

下面做了我需要的。

mask = ~(np.vstack(strut[i].rpsi[:,0]) <= time.time() - apsiTimeWindow)

这将仅输出被屏蔽的col 0。