Функция equal

Функция equal выполняет поэлементное сравнение двух массивов и возвращает массив логических значений True или False. Первый и второй параметры функции принимают массивы для сравнения. Третьим параметром можно передать условие, при котором сравнение не выполняется.

Синтаксис

np.equal(arr1, arr2, [where])

Пример

Давайте сравним два одномерных массива чисел:

import numpy as np

arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([1, 2, 0, 4, 5])
res = np.equal(arr1, arr2)
print(res)

Результат выполнения кода:

[ True True False True True]

Пример

Давайте сравним два двумерных массива:

import numpy as np

arr1 = np.array([[1, 2, 3], [4, 5, 6]])
arr2 = np.array([[1, 2, 0], [4, 0, 6]])
res = np.equal(arr1, arr2)
print(res)

Результат выполнения кода:

[[ True  True False]
 [ True False  True]]

Пример

Сравнение массивов с использованием параметра where, который позволяет выполнять сравнение только для определенных элементов:

import numpy as np

arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([1, 2, 0, 4, 5])
condition = np.array([True, False, True, False, True])
res = np.equal(arr1, arr2, where=condition)
print(res)

Результат выполнения кода:

[ True False False False True]

Пример

Сравнение массивов с числами с плавающей запятой:

import numpy as np

arr1 = np.array([1.0, 2.0, 3.0, 4.0])
arr2 = np.array([1.0, 2.1, 3.0, 4.1])
res = np.equal(arr1, arr2)
print(res)

Результат выполнения кода:

[ True False True False]

Пример

Сравнение массива с числом (вещание):

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
res = np.equal(arr, 3)
print(res)

Результат выполнения кода:

[False False True False False]

Смотрите также

функцию not_equal,
которая выполняет поэлементное сравнение на неравенство
функцию array_equal,
которая проверяет полное равенство массивов
функцию allclose,
которая проверяет приблизительное равенство массивов с допуском
функцию isclose,
которая выполняет поэлементное сравнение с допуском