numpy怎么筛选符合条件的行

可以使用numpy的布尔索引来筛选符合条件的行。

布尔索引是一种使用布尔数组来选择数组中的元素的方法。我们可以通过创建一个布尔数组来指定筛选条件，然后将其与原数组进行索引，以获取符合条件的行。

以下是一个例子：

import numpy as np  # 创建一个示例数组 arr = np.array([[1, 2, 3],                 [4, 5, 6],                 [7, 8, 9]])  # 创建一个布尔数组，指定筛选条件 condition = arr > 5  # 使用布尔索引来筛选符合条件的行 filtered_arr = arr[condition]  print(filtered_arr) 

输出结果为：

[6 7 8 9] 

在上面的例子中，我们创建了一个二维数组arr，并创建了一个布尔数组condition来指定筛选条件，即arr中的元素大于5。然后，我们使用布尔索引arr[condition]来获得满足条件的元素，并将其保存在filtered_arr中。最后，我们打印出filtered_arr，即符合条件的行。