只有一列的numpy数组掩码

Question

我只想在我的2D数组strut[i].rpsi中屏蔽COL0。下面的代码正确地屏蔽了col0，但也屏蔽了col1。

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)

这将只输出COL0 masked。