Сравнение элементов массивов NumPy в Python

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:

или есть более чистый способ сделать это?

Iиспользую Python v2.6

Answer 1

Единственный способ - использовать np.allclose :

np.allclose(a,b)

Хотя для целых чисел

not (a-b).any()

быстрее.

Answer 2

Вы также должны получить сообщение:

Используйте a.any () или a.all ()

Это означает, что вы можете делать следующее:

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

Это потому, что сравнение 2 массивов или массива со списком приводит к логическому массиву. Что-то вроде:

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

Answer 3

преобразовать в список:

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

Answer 4

вы можете попробовать:

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

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