Проверка если матрица симметрична в пакете 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)

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

2 ответов


Вы можете просто использовать allclose

def check_symmetric(a, tol=1e-8):
    return numpy.allclose(a, a.T, atol=tol)

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

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