如何通过某些行/列列表对csr_matrix进行切片?

时间:2018-11-29 05:19:03

标签: python numpy scipy sparse-matrix

假设我的csr_matrix是

I = [0, 1, 3 ,4 ,5, 0, 3]
J = [1, 3 , 4, 9 ,1 , 0, 8]
V = [1]*6
V.append(0)
data=sparse.coo_matrix((V,(I,J)),shape=(7,10)).tocsr()

输出

(0, 0)  1
(0, 1)  1
(1, 3)  1
(3, 4)  1
(3, 8)  0
(4, 9)  1
(5, 1)  1

因此,我想获取具有特定行/列列表的条目。例如:

print(data[:,[3,5,8]])
(1, 0) 1

但是我希望得到,

(1, 3)    1
(3, 8)    0

类似地,我想获取特定的行,但是data [list:]无法提供所需的输出。我该怎么解决?

0 个答案:

没有答案