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

Пользователь

от deion , в категории: Python , 3 месяца назад

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

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

Пользователь

от marisa , 2 месяца назад

@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() или оператор == как показано выше.