Сравнение массивов

Коротко

• a == b, a < b и другие сравнения работают поэлементно и возвращают массив True/False, а не одно значение.
• Поэтому if a == b: падает: Python не знает, как превратить массив в одно True или False.
• Равенство целиком — np.array_equal(a, b): одинаковые форма и все элементы.
• Дробные числа сравнивают с допуском: np.allclose (целиком) и np.isclose (поэлементно).
• NaN не равен ничему, даже себе; для сравнения с учётом NaN есть equal_nan=True.

Синтаксис

a == b                         # поэлементно → массив bool
(a == b).all()                 # все ли совпали (формы должны совпадать)
np.array_equal(a, b, equal_nan=False)
np.allclose(a, b, rtol=1e-05, atol=1e-08, equal_nan=False)
np.isclose(a, b, rtol=1e-05, atol=1e-08)   # поэлементно

Параметры

Условие близости: |a − b| <= atol + rtol × |b|.

Примеры

plan = np.array([100, 120, 90])   # план продаж
fact = np.array([100, 115, 95])   # факт

print(plan == fact)
fact >= plan

[ True False False]
array([ True, False,  True])

print(np.array_equal(plan, fact))
print(np.array_equal(plan, [100, 120, 90]))
np.array_equal(plan, [100, 120])          # другая форма — просто False

False
True
False

x = np.array([0.1 + 0.2, 1 / 3 * 3])
print(x == np.array([0.3, 1.0]))
print(np.isclose(x, [0.3, 1.0]))
np.allclose(x, [0.3, 1.0])

[False  True]
[ True  True]
True

measured = np.array([9.98, 20.03])
print(np.allclose(measured, [10, 20]))              # по умолчанию допуск очень мал
np.allclose(measured, [10, 20], atol=0.05)

False
True

Подводные камни

if plan == fact:
    print("план выполнен")

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

a = np.array([1.0, np.nan])
print(a == a)
print(np.array_equal(a, a))
np.array_equal(a, a, equal_nan=True)

[ True False]
False
True