Как сравнить массивы в numpy?

@deion 

В numpy есть функция numpy.array_equal(), которая может использоваться для сравнения двух массивов. Она возвращает True, если массивы равны и False в противном случае.

1
2
3
4
5
6
7

import numpy as np
a = np.array([1, 2, 3])
b = np.array([1, 2, 3])
c = np.array([4, 5, 6])

print(np.array_equal(a, b))  # True
print(np.array_equal(a, c))  # False

Вы также можете сравнивать массивы используя оператор ==

1
2
3
4
5
6
7

import numpy as np
a = np.array([1, 2, 3])
b = np.array([1, 2, 3])
c = np.array([4, 5, 6])

print((a == b).all())  # True
print((a == c).all())  # False

Если вам нужно сравнивать содержимое массивов по-элементно, вы можете использовать метод numpy.array_equal() или оператор == как показано выше.