modo efficiente per creare una matrice sparsa diagonale

Question

Ho il seguente codice in Python utilizzando Numpy:

p = np.diag(1.0/np.array(x)) 

Come posso trasformarlo per ottenere la matrice sparsa p2 con gli stessi valori di p senza creare p prima?

Answer 1

Utilizzare scipy.sparse.spdiags (che fa molto, e quindi può essere fonte di confusione, all'inizio), scipy.sparse.dia_matrix e/o scipy.sparse.lil_diags. (a seconda del format si desidera la matrice sparse in ...)

E.g. utilizzando spdiags:

import numpy as np 
import scipy as sp 
import scipy.sparse 

x = np.arange(10) 

# "0" here indicates the main diagonal... 
# "y" will be a dia_matrix type of sparse array, by default 
y = sp.sparse.spdiags(x, 0, x.size, x.size) 

Answer 2

Utilizzando il modulo scipy.sparse,

p = sparse.dia_matrix(1.0/np.array(x), shape=(len(x), len(x)));