Python 3.9中如何筛选3列数组第三列小于5的数值

你写出的筛选逻辑存在两处关键偏差，直接导致空数组输出：

• 取判断依据时选错了轴：data[2] < 5 实际提取的是行索引为2的第三整行的元素做判断，全程没有读取第三列的数值做筛选依据。比如你示例里第三行的三个值是18、30、31，所以这个判断会生成全为False的长度为3的布尔数组。
• 布尔索引位置放错：把生成的布尔数组放在列索引位时，原本是用来筛选符合条件的列的；由于刚才生成的布尔数组全为False，会判定所有列都不满足筛选条件，最终匹配到0个有效列，才会输出形状为(258, 0)的空数组。

根据你“仅打印数组第三列中数值小于5的元素”的要求，可直接使用如下写法：

# 步骤拆解：
# 1. data[:, 2] 取出所有行的第三列，生成和行数等长的布尔掩码，标记每行第三列的值是否小于5
# 2. 将掩码放在行索引位置，筛选所有满足条件的行
# 3. 在列索引位置取第三列，直接得到目标值
result = data[data[:, 2] < 5, 2]
print(result)

用你给出的示例数组测试，输出结果为array([2, 2])，完全符合预期。

如果后续需要保留满足筛选条件的整行数据（而非只提取第三列的目标值），可以简化为：

result = data[data[:, 2] < 5]

二维numpy数组的索引规则是用逗号分隔行、列的筛选规则：

• 取所有行的某一列：data[:, 列索引]
• 取所有列的某一行：data[行索引, :]
  不要把行、列的索引位置写混。

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