如何使用NumPy筛选并输出多维数组中首列大于2的元素

直接用NumPy的布尔索引就能快速实现这个需求，具体操作如下：

• 导入numpy库：

import numpy as np

• 创建示例数组：

arr = np.array([[2, 55, 62], [3, 56, 63], [4, 57, 64]])

• 执行筛选操作：

filtered_arr = arr[arr[:, 0] > 2]

关键逻辑说明 #

• arr[:, 0] 提取数组所有行的首列元素
• arr[:, 0] > 2 生成一个布尔数组，对应每一行是否满足“首列大于2”的条件
• 将这个布尔数组作为索引传入原数组，即可直接筛选出符合条件的完整行

运行后得到的filtered_arr结果为：

array([[ 3, 56, 63],
       [ 4, 57, 64]])

内容的提问来源于stack exchange，提问作者Welton