1. 程式人生 > >scipy: 稀疏矩陣 indptr

scipy: 稀疏矩陣 indptr

例子:

from scipy import sparse
data = np.array([1, 4, 5, 2, 3, 6])
indices = np.array([0, 2, 2, 0, 1, 2])
indptr = np.array([0, 2, 3, 6])
mtx = sparse.csr_matrix((data, indices, indptr), shape=(3, 3))

結果:
matrix([[1, 0, 4],
[0, 0, 5],
[2, 3, 6]])

  • indptr[0] = 0 、indptr[1]=2,所以有indices[0:2]=0,2,則第一行的位置0、2對應data中的1、4;
  • 以此類推,indptr[1]=2 、 indptr[2]=3, indices[2:3]=2,則第二行的位置2處對應data中的5.
  • 。。。。