2014-10-20 42 views
1

我有一个稀疏矩阵,说:翻转行

from scipy import sparse 
a = sparse.lil_matrix((2,3),) 
a[0] = [1, 2, 3] 
a[1, 2] = 5 

所以它看起来像:

(0, 0) 1.0 
(0, 1) 2.0 
(0, 2) 3.0 
(1, 2) 5.0 

我想知道 - 有一种简单的方法来翻转行(类似numpy.fliplr等效)? ...所以我会得到输出:

(0, 0) 3.0 
(0, 1) 2.0 
(0, 2) 1.0 
(1, 0) 5.0 

回答

0

一种方法是将数组转换为csr格式,然后操纵行索引:

from scipy import sparse 
a = sparse.lil_matrix((2,3),) 
a[0] = [1, 2, 3] 
a[1, 2] = 5 

a = a.tocsr() 
a.indices = -a.indices + a.shape[1] - 1 
print(a) 

产量

(0, 2) 1.0 
    (0, 1) 2.0 
    (0, 0) 3.0 
    (1, 0) 5.0 
0

你可以做多指标分配:

ii = [2,1,0,3] 
a[:,1] = a[ii,:] 

其中ii是您必须以某种方式创建的索引数组。