← retour aux snippets

scipy.sparse: CSR opérations

Créer et manipuler des matrices creuses CSR.

python stats #scipy#sparse#csr

objectif

Créer et manipuler des matrices creuses CSR.

code minimal

import numpy as np
from scipy.sparse import csr_matrix
A = csr_matrix(([1,2,3], ([0,1,2], [0,1,2])), shape=(3,3))
print((A.T @ A).shape)

utilisation

from scipy.sparse import csr_matrix
B = csr_matrix([[0,1],[2,0]])
print(B.todense().shape)

variante(s) utile(s)

from scipy.sparse import diags
print(diags([1,2,3]).shape[0])

notes

  • CSR: rapide pour produits et accès par lignes.