將值添加到包含索引和值的Scipy稀疏矩陣

Question

我正在進行電力系統分析的程序，我需要使用稀疏矩陣。

有一個常規的地方我填一個稀疏矩陣只是下面的調用：

self.A = bsr_matrix((val, (row,col)), shape=(nele, nbus), dtype=complex) 

由於這個矩陣不會隨時間而改變。另一個矩陣會隨着時間而改變，我需要更新它。有沒有一種方式，例如具有：

co  = [ 2, 3, 6] 
row = [ 5, 5, 5] 
val = [ 0.1 + 0.1j, 0.1 - 0.2j, 0.1 - 0.4j] 

我可以添加那些先前初始化稀疏矩陣？怎麼會更pythonic的方式來做到這一點？

謝謝

Answer 1

您應該使用coo_matrix相反，在那裏你可以改變屬性先前創建的稀疏矩陣的col，row和data：

from scipy.sparse import coo_matrix 
nele=30 
nbus=40 
col = [ 2, 3, 6] 
row = [ 5, 5, 5] 
val = [ 0.1 + 0.1j, 0.1 - 0.2j, 0.1 - 0.4j] 
test = coo_matrix((val, (row,col)), shape=(nele, nbus), dtype=complex) 

print test.col 
#[2 3 6] 
print test.row 
#[5 5 5] 
print test.data 
#[ 0.1+0.1j 0.1-0.2j 0.1-0.4j]