如何在 Python 计算中对向量求平方

Question

假设您有一个带有向量的代码，并且想要对向量进行平方：

import numpy as np
import sympy as sp

a, b, c, f, g = sp.symbols('a b c f g')

M = np.array([a, b, c])

现在你有一个公式，你需要像这样平方这个向量：

B = f * M**2 * g

你如何做到这一点，因为

np.vectorize(sp.simplify)(M**2)

似乎不起作用。

Answer 1

将其保存在 Sympy 内。我们可以通过与自身进行点积来获得平方范数：

a, b, c, f, g = sp.symbols('a b c f g')

M = sp.Matrix([a, b, c])

B = f*(M.dot(M))*g