np.array筛选出符合条件的行数据

可以使用布尔索引来筛选出符合条件的行数据。假设有一个二维数组arr，需要筛选出所有第一列元素大于等于10的行数据，可以使用以下代码：

import numpy as np

arr = np.array([[10, 20, 30], [5, 15, 25], [12, 18, 22]])

# 使用布尔索引筛选出符合条件的行数据
mask = arr[:, 0] >= 10
result = arr[mask]

print(result)

输出结果为：

[[10 20 30]
 [12 18 22]]

其中，mask = arr[:, 0] >= 10 表示对arr数组的第一列进行比较，生成一个布尔数组，True表示该行第一列元素大于等于10，False表示小于10。然后使用arr[mask]将符合条件的行数据筛选出来。