python coo

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