0
我有一个scipy CSR矩阵,我想获取每行的元素列索引。我的做法是:稀疏矩阵:如何获得每行的非零索引
import scipy.sparse as sp
N = 100
d = 0.1
M = sp.rand(N, N, d, format='csr')
indM = [row.nonzero()[1] for row in M]
INDM正是我需要的,它有相同数量的行为M,看起来像这样:
[array([ 6, 7, 11, ..., 79, 85, 86]),
array([12, 20, 25, ..., 84, 93, 95]),
...
array([ 7, 24, 32, 40, 50, 51, 57, 71, 74, 96]),
array([ 1, 4, 9, ..., 71, 95, 96])]
的问题是,与大矩阵这种方法看起来很慢。 有什么办法避免列表理解或以某种方式加快速度?
谢谢。
大,谢谢! –
'M.tolil()。rows'给出一个具有相同值列表的对象数组。 – hpaulj