对 numpy 数组进行 if 条件检查它是否等于一个值

Question

我有一个1D数组，它根据轮数增加，首先从

开始

a = np.array([5])  # round 1 
a = np.array([5, 5]) # round 2 
a = np.array([5, 5, 5]) # round 3
...

如何检查任意轮数，以便每次运行它时，它都会检查

if a == 5:       # pseudo if a = np.array([5, 5]) == 5
# do something 

有办法吗?

Answer 1

Python的all运算符处理这个

if all(a == 5):