from scipy.sparse import csr_matrix
arr = np.array([[0,1,0,2,0],[1,1,0,2,0],[2,0,5,0,0]])
b = csr_matrix(arr)
#print(b.shape) # (3, 5)
#print(b.nnz) #非零个数, 7
print(b.data) #非零值, [1 2 1 1 2 2 5]
print('\n')
print(b.indices) # #稀疏矩阵非0元素对应的列索引值所组成数组, [1 3 0 1 3 0 2]
print('\n')
print(b.indptr) #第一个元素0,之后每个元素表示稀疏矩阵中每行元素(非零元素)个数累计结果, [0 2 5 7]
print('\n')
print(b.toarray()) # [[0,1,0,2,0],[1,1,0,2,0],[2,0,5,0,0]]
csr_matrix的用法--shape, nnz, data, indices, indptr, toarray
最新推荐文章于 2022-09-11 16:35:49 发布