python coo_matrix取某一行或者某一列

最近用到了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)

参考文献

scipy.sparse.coo_matrix