python 的csr_python – 设置csr_matrix行

最后,我设法通过索引杂耍完成了这项工作.

def set_row_csr(A, row_idx, new_row):

'''

Replace a row in a CSR sparse matrix A.

Parameters

----------

A: csr_matrix

Matrix to change

row_idx: int

index of the row to be changed

new_row: np.array

list of new values for the row of A

Returns

-------

None (the matrix A is changed in place)

Prerequisites

-------------

The row index shall be smaller than the number of rows in A

The number of elements in new row must be equal to the number of columns in matrix A

'''

assert sparse.isspmatrix_csr(A), 'A shall be a csr_matrix'

assert row_idx < A.shape[0], \

'The row index ({0}) shall be smaller than the number of rows in A ({1})' \

.format(row_idx, A.shape[0])

try:

N_elements