"如何按条件提取 2D NumPy 数组的行？"

Question

我有一个 4*5 NumPy 数组，我想检索所有元素都小于 5 的行。

arr = np.array([[0,2,3,4,5],[1,2,4,1,3], [2,2,5,4,6], [0,2,3,4,3]])
arr[np.where(arr[:,:] <= 4)] 

预期输出：

[[1,2,4,1,3],[0,2,3,4,3]]

实际输出：

array([0, 2, 3, 4, 1, 2, 4, 1, 3, 2, 2, 4, 0, 2, 3, 4, 3])

任何帮助表示赞赏！

Answer 1

这其实很简单。 只需将整个数组转换为布尔值（如果小于 5，则每个值为 True，否则为 False），然后使用带有axis=1<\/code>的np.all<\/code>为所有项目为 True 的每一行返回 True：

>>> arr[np.all(arr < 5, axis=1)]
array([[1, 2, 4, 1, 3],
       [0, 2, 3, 4, 3]])