如何使用 numpy 检查每行每个元素的条件

Question

对于机器学习，我应用 Parzen Window 算法。

我有一个数组 (m,n)。我想检查每一行是否有任何值 > 0.5，如果每个值都 > 0.5，那么我将返回 0，否则返回 1。

我想知道是否有一种方法可以在不使用 numpy 循环的情况下完成此操作。

Answer 1

您可以在布尔数组上使用np.allwith 。axis=1

import numpy as np
arr = np.array([[0.8, 0.9], [0.1, 0.6], [0.2, 0.3]])

print(np.all(arr>0.5, axis=1))
>> [True False False]