ndarray 的联合条件截断答案

【问题标题】：joint conditional truncation for ndarrayndarray 的联合条件截断
【发布时间】：2023-03-21 22:52:01
【问题描述】：

给定一个 2-dim numpy 数组

array([[-1.00, 1.00 ],
       [-2.00, 2.00 ],
       [ 2.00, 3.00 ],
       [ 1.00, 3.00 ],
       [-4.00,-3.00]])

我希望检索都超过给定值的行。也就是说，说所有大于或等于一的值。这将是所需的输出。

array([[ 2.00, 3.00 ],
       [ 1.00, 3.00 ]])

对于一维数组，可以直接这样完成

sample[sample >=1]

但是，它不适用于 dim > 1 的数组。上面的语法也适用于 pandas 数据帧，但我不想将所有内容都转换为数据帧以进行截断。

有没有一种优雅的方式来做到这一点？谢谢你的帮助。

【问题讨论】：

【解决方案1】：

你可以使用np.array.all(axis=1):

>>> sample[(sample >= 1).all(axis=1)]
array([[2., 3.],
       [1., 3.]])

【讨论】：