Dodaj wartości do rzadkiej macierzy Scipy z indeksami i wartościami

Question

Pracuję w programie do analizy Power System i muszę pracować z rzadkimi macierzami.

Jest rutynowe gdzie wypełnić rzadki matrycy tylko z poniższej rozmowy:

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

W tej macierzy nie zmienia się w czasie. Kolejna macierz zmienia się z czasem i muszę ją zaktualizować. Czy istnieje sposób, że mając na przykład:

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

mogę dodać te do wcześniej inicjowane rozrzedzony matrycy? Jak byłoby to bardziej pythonic sposób to zrobić?

Dziękuję

Answer 1

Należy użyć coo_matrix zamiast, gdzie można zmieniać atrybuty col, row i data z wcześniej utworzonego rozrzedzony matrycy:

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]