比较python中numpy数组的元素答案

【问题标题】：Comparing elements of numpy arrays in python比较python中numpy数组的元素
【发布时间】：2011-07-08 18:29:37
【问题描述】：

我想比较两个 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

【问题讨论】：

【解决方案1】：

numpy的方式是使用np.allclose:

np.allclose(a,b)

虽然对于整数，

not (a-b).any()

更快。

【讨论】：

【解决方案2】：

你也应该得到消息：

使用 a.any() 或 a.all()

这意味着您可以执行以下操作：

if (output[x][y] != [150,25,75]).all():

这是因为将 2 个数组或一个数组与一个列表进行比较会产生一个布尔数组。比如：

array([ True,  True,  True], dtype=bool)

【讨论】：

【解决方案3】：

转换为列表：

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

【讨论】：

【解决方案4】：

你可以试试：

a = output[x][y]
b = [150,25,75]

if not all([i == j for i,j in zip(a, b)]):

【讨论】：