在`scipy`元素方面有效地反轉`csr_matrix`

Question

令 $A$ 為csr_matrix表示圖的連通性矩陣，其中 $A_{ij}$ 是邊的權重。 現在，我需要以有效的方式反轉矩陣的每個非零元素。 我現在這樣做的方式是

B = 1.0 / A.toarray()
B[B == np.inf] = 0

這有兩個缺點：

1. 通過將 csr_matrix 轉換為數組來增加內存使用量。
2. 除以零發生

有什么建議可以更有效地做到這一點嗎？

Answer 1

你可以做到這一點的方法之一是創建從一個新的矩陣data ， indices和indptr的A ： B = csr_matrix((1/A.data, A.indices, A.indptr))

（這假設A中沒有明確存儲的零，因此1/A.data不會導致某些值成為inf 。）

例如，

In [108]: A
Out[108]: 
<4x4 sparse matrix of type '<class 'numpy.float64'>'
    with 4 stored elements in Compressed Sparse Row format>

In [109]: A.A
Out[109]: 
array([[0. , 1. , 2.5, 0. ],
       [0. , 0. , 0. , 0. ],
       [0. , 0. , 0. , 4. ],
       [2. , 0. , 0. , 0. ]])

In [110]: B = csr_matrix((1/A.data, A.indices, A.indptr))

In [111]: B
Out[111]: 
<4x4 sparse matrix of type '<class 'numpy.float64'>'
    with 4 stored elements in Compressed Sparse Row format>

In [112]: B.A
Out[112]: 
array([[0.  , 1.  , 0.4 , 0.  ],
       [0.  , 0.  , 0.  , 0.  ],
       [0.  , 0.  , 0.  , 0.25],
       [0.5 , 0.  , 0.  , 0.  ]])

Answer 2

csr有一個power方法：

In [598]: M = sparse.csr_matrix([[0,3,2],[.5,0,10]])
In [599]: M
Out[599]: 
<2x3 sparse matrix of type '<class 'numpy.float64'>'
    with 4 stored elements in Compressed Sparse Row format>
In [600]: M.A
Out[600]: 
array([[ 0. ,  3. ,  2. ],
       [ 0.5,  0. , 10. ]])
In [601]: x = M.power(-1)
In [602]: x
Out[602]: 
<2x3 sparse matrix of type '<class 'numpy.float64'>'
    with 4 stored elements in Compressed Sparse Row format>
In [603]: x.A
Out[603]: 
array([[0.        , 0.33333333, 0.5       ],
       [2.        , 0.        , 0.1       ]])