如何检查NumPy一维数组所有元素是否均等于指定值？

可以利用NumPy的向量化特性结合all()方法快速实现，逻辑简单且高效，适配任意长度的一维数组：

import numpy as np

# 当前轮次的目标数组
a = np.array([5, 5, 5])  # 替换为对应轮次的实际数组

# 核心检查逻辑
if (a == 5).all():
    # 此处编写需要执行的操作
    print("所有元素均为5，执行指定操作")

逻辑说明 #

• a == 5会对数组每个元素做相等性判断，返回一个同形状的布尔数组，每个元素标记对应位置是否等于5
• .all()方法会校验布尔数组内所有元素是否全为True，全满足时返回True，否则返回False

补充提示 #

这种方法基于NumPy底层优化的向量化操作，比Python原生循环遍历检查的效率高得多，即使是大规模数组也能快速完成判断。你的场景中数组长度随轮次递增（至少为1），无需额外处理空数组的边界情况。

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