如何编写循环遍历NumPy矩阵每行第一列并按条件删除对应行

第一段代码存在的问题 #

• 循环逻辑冗余混乱：while循环和遍历第一列的for循环嵌套完全没有必要，且边删除行边用固定的n作为索引，数组长度变化后索引会错位，很容易删错数据或者触发索引越界错误
• 条件判断逻辑错误：if a[:,0] not in range(1,30)是直接判断整个第一列数组是否在range范围内，而非判断单个元素是否符合要求；且range(1,30)本身是左闭右开区间，仅包含1到29的整数，和需求的(1,30)开区间也不匹配
• 循环边界错误：while循环条件设置为n<=len(a)，数组合法索引范围是0 ~ len(a)-1，n最终会等于数组长度，触发索引越界

第二段代码存在的问题 #

• 遍历逻辑错误：for row in range(len(a))拿到的row是整数类型的行索引，后续for element in row相当于遍历整数，直接会触发语法报错
• 条件完全不符合需求：判断条件写的是element not in range(1,6)，和要求的「第一列元素处于(1,30)范围」完全不匹配
• 同样存在边删行边用固定索引的问题，数组长度变化后n和行的对应关系完全错乱，删除逻辑必然出错

推荐写法（NumPy向量化操作，无需循环） #

NumPy原生支持布尔索引筛选，不需要写循环，执行效率更高，也不会出现索引错位问题：

import numpy as np
a = np.array([[1,2,3],[10,20,30],[40,50,60]])
# 此处按需求开区间(1,30)判断，若需要闭区间/左闭右开，自行修改判断符号即可
mask = (a[:, 0] > 1) & (a[:, 0] < 30)
a = a[mask]
print(a)

输出结果为：

[[10 20 30]]

循环写法（仅作逻辑参考） #

如果必须用循环实现，不要边遍历边删除原数组，应该先收集需要保留的行索引，最后一次性筛选：

import numpy as np
a = np.array([[1,2,3],[10,20,30],[40,50,60]])
keep_idx = []
for i in range(a.shape[0]):
    first_col = a[i, 0]
    if 1 < first_col < 30:
        keep_idx.append(i)
a = a[keep_idx]
print(a)

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