查找numpy 的array中 满足特定条件的行并且返回这些行

可以使用布尔索引来查找满足特定条件的行，并且返回这些行。假设有一个二维的numpy数组arr，要查找第一列元素大于等于5的行，可以使用以下代码：

import numpy as np

arr = np.array([[1, 2, 3], [5, 6, 7], [8, 9, 10], [11, 12, 13]])

condition = (arr[:, 0] >= 5)  # 第一列元素大于等于5的布尔数组
result = arr[condition]  # 返回满足条件的行

print(result)

输出结果为：

[[ 5  6  7]
 [ 8  9 10]
 [11 12 13]]

其中，condition是一个布尔数组，表示第一列元素是否大于等于5。这里使用了切片操作arr[:, 0]来取出第一列元素，并进行比较。然后使用condition作为索引来返回满足条件的行。