【Sage数学库】计算符号矩阵的平方、立方

p = var('p')
q = var('q')
# q = 1-p
m_symbol = matrix( [ [1, 0, 0, 0], [q, 0, p, 0], [0, q, 0, p], [0, 0, 0, 1] ] )
print("【齐次马尔可夫链】用符号p、q表示的【1步转移概率矩阵】m_symbol：")
print(m_symbol)
print("【齐次马尔可夫链】用符号p、q表示的【2步转移概率矩阵】m_2: ")
m_2 = m_symbol ** 2
print(m_2)
print("【齐次马尔可夫链】用符号p、q表示的【3步转移概率矩阵】m_3: ")
m_3 = m_symbol ** 3
print(m_3)

【齐次马尔可夫链】用符号p、q表示的【1步转移概率矩阵】m_symbol：
[1 0 0 0]
[q 0 p 0]
[0 q 0 p]
[0 0 0 1]
【齐次马尔可夫链】用符号p、q表示的【2步转移概率矩阵】m_2:
[  1     0        0         0    ]
[  q     p*q     0        p^2 ] 
[q^2   0      p*q       p     ]
[  0     0       0         1     ]
【齐次马尔可夫链】用符号p、q表示的【3步转移概率矩阵】m_3:
[        1               0             0            0             ]
[p*q^2 + q          0