CODEWITHSUNDEEP
matrixvectorsympy
John Forbes
John Forbes

Reputation: 1354

How to perform matrix by vector multiplication with sympy?

I have:

I would like to multiply the matrix by the vector in order to produce a vector.

Can I perform this operation conveniently or do I need to do some extra manipulation first?

For reference I am attempting to get the symbolic result of a rotation matrix applied to a vector.

Also below, is some of my code that deals with the above matrix and vector.

from sympy.vector import CoordSys3D

σ, θ, γ, λ, a, b, c = symbols('σ, θ, γ, λ, a, b, c, a_v, b_v, c_v')
σ = sin(θ)
γ = cos(θ)
λ = 1 - γ

N = CoordSys3D('N')
u = a*N.i + b*N.j + c*N.k # Axis of rotation

R = Matrix([
    [a*a*λ + γ, a*b*λ-c*σ, a*c*λ+b*σ],
    [b*a*λ+c*σ, b*b*λ + γ, b*c*λ-a*σ],
    [c*a*λ-b*σ, c*b*λ+a*σ, c*c*λ + γ],
])

# Input vector prior to rotation
v = a_v*N.i + b_v*N.j + c_v*N.k

# How to calculate the post rotation output vector w = Rv?

In summary is there a built-in mechanism in sympy for matrix by vector multiplication?

Upvotes: 6

Views: 3967

Answers (2)

Zciurus
Zciurus

Reputation: 838

In the current version of SymPy (1.11), you can calculate the vector matrix product by using the matmul operator (@)

The following code works for me:

v = Matrix([x, y, z])
Kx = Matrix([[1,        0,        0      ], 
             [0,        cos(kx), -sin(kx)], 
             [0,        sin(kx),  cos(kx)]])

product = Kx @ v


# Don't:
# product = v @ Kx

Upvotes: 2

John Forbes
John Forbes

Reputation: 1354

Although I didn't find a function to do what I wanted, this code achieved the same result. I'm posting it here in case it is useful for others.

w = R * Matrix([v.coeff(N.i), v.coeff(N.j), v.coeff(N.k)])

Upvotes: 4

Related Questions