Проверка симметричности матрицы в Numpy

Я пытаюсь сделать функцию с аргументами (a,tol=1e-8) которая возвращает логическое значение, которое сообщает пользователю, симметрична ли матрица (симметричная матрица равна ее транспонированию). Пока у меня есть:

def check_symmetric(a, tol=1e-8):
if np.transpose(a, axes=axes) == np.transpose(a, axes=axes):
    return True
def sqr(s):
    rows = len(s)
    for row in sq:
        if len(row) != rows:
            return False
    return True
if a != sqr(s):
    raise ValueError

хотя я продолжаю получать axes isn't defined сообщение, поэтому я уверен, что это не работает... тесты, которые я хотел бы пройти, это:

e = np.eye(4)
f = np.diag([1], k=3)
g = e[1:, :]

print(check_symmetric(e))
print(not check_symmetric(e + f))
print(check_symmetric(e + f * 1e-9))
print(not check_symmetric(e + f * 1e-9, 1e-10))
try:
    check_symmetric(g)
    print(False)
except ValueError:
    print(True)

Любая помощь приветствуется, спасибо!

Ответы

Ответ 1

Вы можете просто сравнить его с транспонированием, используя allclose

def check_symmetric(a, rtol=1e-05, atol=1e-08):
    return numpy.allclose(a, a.T, rtol=rtol, atol=atol)

Ответ 2

Следующая функция также решает проблему:

def check_symmetric(a, tol=1e-8):
    return np.all(np.abs(a-a.T) < tol)