最近用到了sparse里面的coo_matrix结构,结果它并不能像以前的list或者numpy根据索引来取某一行或者某一列,那怎么办呢?
from scipy import sparse
import numpy as np
matrix = np.array([[9, 8, 7], [6, 5, 4], [3, 2, 1]])
mat= sparse.coo_matrix(matrix)
print(mat.getcol(1))
print(mat.getrow(1))
用getrow和getcol就可以了,我的执行记录为:
>>> mat.getcol(1).todense()
matrix([[8],
[5],
[2]], dtype=int64)
>>> mat.getrow(1).todense()
matrix([[6, 5, 4]], dtype=int64)