sp.coo_matrix() 详解

Posted 2021-10-08 K同学啊

sp.coo_matrix() 的作用是生成矩阵

1. 构造一个空矩阵

这种用法比较简单，直接生成一个给定维度的矩阵，并且可以指定元素类型。如下指定生成一个(3,4)且数据类型为int16的矩阵

实例：

import scipy.sparse as sp
import numpy as np

A = sp.coo_matrix((3, 4), dtype=np.int16).toarray()
A
"""
输出：
array([[0, 0, 0, 0],
       [0, 0, 0, 0],
       [0, 0, 0, 0]], dtype=int16)
"""

2.用指定数据生成矩阵

实例：

import scipy.sparse as sp
import numpy as np

row  = np.array([0, 3, 1, 0])
col  = np.array([0, 3, 1, 2])
data = np.array([4, 5, 7, 9])
B = sp.coo_matrix((data, (row, col)), shape=(4, 4)).toarray()
B
"""
array([[4, 0, 9, 0],
       [0, 7, 0, 0],
       [0, 0, 0, 0],
       [0, 0, 0, 5]])
"""