交换法排序matlab,matlab – 通过交换行和列重新排列稀疏数组

CSC格式保留所有非零条目的行索引列表,CSR格式保留所有非零条目的列索引列表.我认为你可以利用这个来交换周围的东西,我认为不应该有任何副作用：

def swap_rows(mat, a, b) :

mat_csc = scipy.sparse.csc_matrix(mat)

a_idx = np.where(mat_csc.indices == a)

b_idx = np.where(mat_csc.indices == b)

mat_csc.indices[a_idx] = b

mat_csc.indices[b_idx] = a

return mat_csc.asformat(mat.format)

def swap_cols(mat, a, b) :

mat_csr = scipy.sparse.csr_matrix(mat)

a_idx = np.where(mat_csr.indices == a)

b_idx = np.where(mat_csr.indices == b)

mat_csr.indices[a_idx] = b

mat_csr.indices[b_idx] = a

return mat_csr.asformat(mat.format)

你现在可以这样做：

>>> mat = np.zeros((5,5))

>>> mat[[1, 2, 3, 3], [0, 2, 2, 4]] = 1

>>> mat = scipy.sparse.lil_matrix(mat)

>>> mat.todense()

matrix([[ 0., 0., 0., 0., 0.],

[ 1., 0., 0., 0., 0.],

[ 0., 0., 1., 0., 0.],

[ 0., 0., 1., 0., 1.],

[ 0., 0., 0., 0., 0.]])

>>> swap_rows(mat, 1, 3)

<5x5 sparse matrix of type ''

with 4 stored elements in LInked List format>

>>> swap_rows(mat, 1, 3).todense()

matrix([[ 0., 0., 0., 0., 0.],

[ 0., 0., 1., 0., 1.],

[ 0., 0., 1., 0., 0.],

[ 1., 0., 0., 0., 0.],

[ 0., 0., 0., 0., 0.]])

>>> swap_cols(mat, 0, 4)

<5x5 sparse matrix of type ''

with 4 stored elements in LInked List format>

>>> swap_cols(mat, 0, 4).todense()

matrix([[ 0., 0., 0., 0., 0.],

[ 0., 0., 0., 0., 1.],

[ 0., 0., 1., 0., 0.],

[ 1., 0., 1., 0., 0.],

[ 0., 0., 0., 0., 0.]])

我使用了LIL矩阵来展示如何保留输出的类型.在您的应用程序中,您可能希望已经采用CSC或CSR格式,并根据它选择是先交换行还是列,以最大限度地减少转换.