python - 带变量的抽象矩阵乘法

Question

我知道 python 进行矩阵乘法的能力。不幸的是我不知道如何抽象地做到这一点？所以不与确定的数字，但有变量。

例子:

M = ( 1   0 ) * ( 1   d )
    ( a   c )   ( 0   1 )

有没有办法定义a、c和d，使得矩阵乘法给我

( 1   d       )
( a   a*d + c )

?

Answer 1

使用 sympy你可以这样做:

>>> from sympy import *
>>> var('a c d A B')
(a, c, d, A, B)
>>> A = Matrix([[1, 0], [a, c]])
>>> A
Matrix([
[1, 0],
[a, c]])
>>> B = Matrix([[1, d], [0, 1]])
>>> B
Matrix([
[1, d],
[0, 1]])
>>> M = A.multiply(B)
>>> M
Matrix([
[1,       d],
[a, a*d + c]])