Существует ли очень быстрый способ дополнить матрицу столбцов Scipy CSR нулями с обеих сторон?

Question

Например, у меня есть матрица столбцов A следующим образом:

A:
[1,
 2, 
 3, 
 4,]

Мне нужно очень быстро дополнить x и y количеством нулей сверху и снизу. Здесь пусть x = 2 и y = 3. поэтому результирующая матрица B после заполнения будет:

B:
[0,
 0,
 1,
 2, 
 3, 
 4,
 0,
 0,
 0]

Теперь проблема в том, что это ОЧЕНЬ большие матрицы столбцов, и их миллионы, и я Знаю, что в CSR-представлении мне просто нужно изменить форму матрицы и индексы. Может ли кто-нибудь помочь?

Answer 1

Заполнение только A.indptr с numpy.pad должно быть достаточно быстрым:

B = csr_matrix((A.data, A.indices, numpy.pad(A.indptr, (x, y), "edge")))