如何基于指定列过滤二维NumPy数组？报错问题求解

你写的arr[2]取的是数组的第三行，由此生成的布尔数组长度为3，但原数组有4行，维度不匹配才触发了索引错误。如果要基于某一列的值过滤行，得先取出对应列的所有行数据，再生成匹配维度的布尔索引。

正确代码示例 #

先定义原数组：

import numpy as np
arr = np.arange(12).reshape((4,3))
print(arr)
# 输出：
# [[ 0  1  2]
# [ 3  4  5]
# [ 6  7  8]
# [ 9 10 11]]

如果要按第三列（索引为2）的值大于5来过滤行，代码如下：

# 生成第三列值大于5的布尔掩码（长度和行数一致）
mask = arr[:, 2] > 5
# 用掩码过滤行
filtered_arr = arr[mask]
print(filtered_arr)
# 输出：
# [[ 6  7  8]
# [ 9 10 11]]

也可以合并成一行代码：

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

这里arr[:,2]的意思是取数组中所有行（:代表全部行）的第3列（索引从0开始），生成的布尔数组长度为4，和原数组的行数完全匹配，就能正确筛选出符合条件的行。

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