如何访问sympy矩阵元素以进行行操作?

时间:2019-06-07 20:35:33

标签: python matrix sympy

我正在寻找一种访问sympy矩阵元素以执行行操作的方法,但似乎无法提出一种方法或找到描述该过程的任何现有文档。

例如,假设我有以下代码:

import sympy as sp
from sympy import *

matrix = sp.Matrix([[3,2,2],[1,2,3]])

我想在第一行和第二列中划分元素,在这种情况下为2。我可以想到的一种真正的方法是执行以下操作:

a = int(matrix.row(0).col(2)[0])
matrix.row(0)/a

但是现在矩阵的第一行是

[3/2,1,1]

,这次我想将行再次除以3/2,这对于我以前的方法不起作用。如何执行这些行操作,以及如何使它们更新原始矩阵? (即,当我将行除以3时,它会更新原始矩阵中的行,而不仅仅是返回一个仅反映更新后的行的单独矩阵)

而且,有没有简单的方法可以用sympy矩阵进行行交换/交换(即r1 <-> r2)?

编辑:

我发现我可以通过简单地使用matrix[row#,:]/matrix[row#,column#]来完成问题的除法部分,但是我仍然不确定如何使该行操作直接反映在原始矩阵中,或者如何进行行处理交换。

2 个答案:

答案 0 :(得分:0)

当我有这样的问题时,我尝试在目录中寻求帮助:

>>> [w for w in dir(Matrix) if 'op' in w and not w.startswith('_')]
[col_op, copy, copyin_list, copyin_matrix, elementary_col_op, elementary_row_op, 
row_op, zip_row_op]

>>> help(Matrix.row_op)
Help on method row_op in module sympy.matrices.dense:

row_op(self, i, f) unbound sympy.matrices.dense.MutableDenseMatrix method
    In-place operation on row ``i`` using two-arg functor whose args are
    interpreted as ``(self[i, j], j)``.
...

>>> help(Matrix.elementary_row_op)
Help on method elementary_row_op in module sympy.matrices.matrices:

elementary_row_op(self, op='n->kn', row=None, k=None, row1=None, row2=None) unbound 
sympy.matrices.dense.MutableDenseMatrix method
    Performs the elementary row operation `op`.

    `op` may be one of

        * "n->kn" (row n goes to k*n)
        * "n<->m" (swap row n and row m)
        * "n->n+km" (row n goes to row n + k*row m)

    Parameters
    ==========

    op : string; the elementary row operation
    row : the row to apply the row operation
    k : the multiple to apply in the row operation
    row1 : one row of a row swap
    row2 : second row of a row swap or row "m" in the row operation
           "n->n+km"

所以看起来两者都可以使用。

>>> m =  Matrix([[3,2,2],[1,2,3]])
>>> m.row_op(0, lambda x, j: x/2)
>>> m
Matrix([
[3/2, 1, 1],
[  1, 2, 3]])
>>> m.row_op(0, lambda x, j: x/(3/2))
>>> m
Matrix([
[1, 2/3, 2/3],
[1,   2,   3]])

>>> m =  Matrix([[3,2,2],[1,2,3]])
>>> m.elementary_row_op('n->kn',row1=0,k=1/3)
Matrix([
[1, 2/3, 2/3],
[1,   2,   3]])

答案 1 :(得分:0)

我仍然是sympy的新手,但是对numpy很有见识。因此,让我们看看sympy的行为是否大致相同。

isympy会话中:

In [67]: M = Matrix([[3,2,2],[1,2,3]])                                                                 

In [68]: M                                                                                             
Out[68]: 
⎡3  2  2⎤
⎢       ⎥
⎣1  2  3⎦

In [69]: M[0,:]             # a row, using a numpy style indexing                                                                                        
Out[69]: [3  2  2]

In [70]: M[0,1]             # an element                                                                           
Out[70]: 2

In [71]: M[0,:]/M[0,1]      # division, producing a new matrix                                                                            
Out[71]: [3/2  1  1]

In [72]: M                  # no change to M                                                                           
Out[72]: 
⎡3  2  2⎤
⎢       ⎥
⎣1  2  3⎦

In [73]: M[0,:]/=M[0,1]     # but with a /= (Python syntax)                                                                           

In [74]: M                                                                                             
Out[74]: 
⎡3/2  1  1⎤
⎢         ⎥
⎣ 1   2  3⎦

In [75]: M[0,:]/=3/2      # again                                                                             

In [76]: M                                                                                             
Out[76]: 
⎡1.0  0.666666666666667  0.666666666666667⎤
⎢                                         ⎥
⎣ 1           2                  3        ⎦

进行了浮点除法;我怀疑使用其他除数可以进行适当的小数除法。

In [83]: M = Matrix([[3,2,2],[1,2,3]])                                                                 

In [84]: M[0,:]/=M[0,1]                                                                                

In [85]: M[0,:]/=Rational(3,2)                                                                         

In [86]: M                                                                                             
Out[86]: 
⎡1  2/3  2/3⎤
⎢           ⎥
⎣1   2    3 ⎦