Предположим, мой 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:] не дает желаемого результата.Как я могу решить это?