python - 比较python中numpy数组的元素

Question

我想比较两个 1x3 数组，例如:

if output[x][y] != [150,25,75]

(output 这里是 3x3x3 所以 output[x][y] 只是 1x3)。

我收到一条错误消息:

ValueError: The truth value of an array with more than one element is ambiguous. 

这是否意味着我需要这样做:

if output[y][x][0] == 150 and output[y][x][1] == 25 and output[y][x][2] == 75:

或者是否有更简洁的方法来做到这一点？

我正在使用 Python v2.6

Answer 1

numpy 的方法是使用 np.allclose :

np.allclose(a,b)

虽然对于整数，

not (a-b).any()

更快。