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