你如何在python中取包含零的矩阵的平方根？

Question

假设我们要计算单位矩阵的平方根，在 python 中你会怎么做？ 我试过 numpy.sqrt() 但它没有用。 有任何想法吗？ 非常感谢！

Answer 1

你可以试试这个：

my_matrix = np.array([[1.0, 3.0], [1.0, 4.0]])

np.sqrt(my_matrix.data)

Answer 2

您可以使用一个名为scipy的包来完成。 请看下面的代码：

from scipy.linalg import sqrtm
my_matrix = np.array([[1.0, 3.0], [1.0, 4.0]])
sqrt_my_matrix = sqrtm(my_matrix)

Answer 3

如果你想明智地做这个元素，你可以使用

import numpy as np
my_matrix = np.array([0, 4, 9])
sqrt_my_matrix = np.sqrt(my_matrix)
print(sqrt_my_matrix)