manera eficiente para crear una matriz dispersa en diagonal

Question

he el siguiente código en Python usando Numpy:

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

Como puedo transformarlo para obtener la matriz dispersa p2 con los mismos valores que p sin crear p primero?

Answer 1

Use scipy.sparse.spdiags (que hace mucho, y puede ser confuso, al principio), scipy.sparse.dia_matrix y/o scipy.sparse.lil_diags. (dependiendo de format desea la matriz dispersa en ...)

P. ej. utilizando 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

Con el módulo de scipy.sparse,

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