如何在Python中循环内部and.array

Question

我有一个 N 维数组 = X，我想检查 X 中的每个值是否大于 0.35 。我写成：-

for number in X:
    if (.35> number):  # Here error occurs
        print (enumerate(number))

但我收到此错误：

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

我想将 X 中小于 0.35 的每个值的索引保存在列表中

Answer 1

使用 NumPy 时，您必须始终努力在向量空间中进行操作。这意味着不写for循环，也不写in等。

对于您目前的情况，您可以这样做：

print(X[X < 0.35])

这将比编写循环快得多。如果您必须将它们打印在不同的行上，您可以：

values = X[X < 0.35]
np.savetxt(sys.stdout, values)