获取Numpy布尔数组中False元素索引报错，求解决方案

错误原因 #

你的代码存在两个关键问题：

• 数组元素是布尔类型的False，但你错误地和字符串'False'做比较，类型不匹配导致元素级比较失败，触发警告。
• 手动循环遍历numpy数组的方式冗余且低效，numpy提供了原生向量化操作来处理这类索引需求。

修复原循环的写法 #

如果要保留循环逻辑，只需修正比较对象，同时注意你的数组是二维结构（每个元素是长度为1的子数组），需要取出子数组里的布尔值：

import numpy as np

A = np.array([[False],
              [False],
              [True],
              [False],
              [False]])

for i in range(len(A)):
    if not A[i][0]:  # 直接判断布尔值取反，也可以用A[i][0] is False
        print(i)

更高效的numpy原生写法 #

推荐用numpy的向量化操作，一行代码就能得到所有目标索引：

import numpy as np

A = np.array([[False],
              [False],
              [True],
              [False],
              [False]])

# 方法1：用np.where取反布尔数组
indices = np.where(~A)[0]
print(indices)  # 输出 [0 1 3 4]

# 方法2：展平数组后直接筛选
indices = np.flatnonzero(A.flatten() == False)
print(indices)  # 输出 [0 1 3 4]

两种方法都能直接得到你想要的索引数组，比循环高效得多，尤其适合处理大规模数组。

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